Searched defs:value_shape (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | fill_op.cc | 36 const TensorShape value_shape = ctx->InputShape(1); variable 41 OP_REQUIRES(ctx, IsLegacyScalar(value_shape), 43 value_shape.DebugString())); 60 if (value_shape.dims() > 0) { 61 CHECK_EQ(value_shape.dims(), 1);
|
H A D | tensor_array_ops.cc | 277 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); 278 ctx->SetOutput(0, b->Reshape(read, value_shape)); 365 const TensorShape value_shape = ctx->InputShape(2); variable 369 TensorShape elem_shape = value_shape; 389 if (status.ok() && num_indices == value_shape.dim_size(0)) { 402 auto slice_dims = value_shape.dim_sizes(); 405 std::vector<int64> value_starts(value_shape.dims(), 0); 406 auto value_ends = value_shape.dim_sizes(); 408 std::vector<int64> value_strides(value_shape.dims(), 1); 499 TensorShape value_shape variable [all...] |
/external/tensorflow/tensorflow/core/ops/ |
H A D | resource_variable_ops.cc | 97 ShapeHandle value_shape = c->input(1); local 100 c->Merge(handle_shape_and_type.shape, value_shape, &unused));
|
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | embedding_lookup_sparse_test.cc | 38 std::initializer_list<int> value_shape) { 49 lookup_shape, value_shape}); 34 EmbeddingLookupSparseOpModel(CombinerType type, std::initializer_list<int> lookup_shape, std::initializer_list<int> indices_shape, std::initializer_list<int> dense_shape_shape, std::initializer_list<int> value_shape) argument
|
H A D | hashtable_lookup_test.cc | 37 std::initializer_list<int> value_shape, 45 BuildInterpreter({lookup_shape, key_shape, value_shape}); 35 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, std::initializer_list<int> key_shape, std::initializer_list<int> value_shape, TensorType type) argument
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | lookup_table_op.cc | 125 TensorShape value_shape() const override { return TensorShape(); } 154 GetNodeAttr(kernel->def(), "value_shape", &value_shape_)); 254 TensorShape value_shape() const override { return value_shape_; } 310 GetNodeAttr(kernel->def(), "value_shape", &value_shape_)); 472 TensorShape value_shape = MaybeVectorizeShape(value_shape_); variable 479 expected_value_shape.AppendShape(value_shape); 494 TensorShape value_shape() const override { return value_shape_; } 673 output_shape.AppendShape(table->value_shape());
|
Completed in 184 milliseconds