Searched defs:value_shape (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dfill_op.cc36 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 Dtensor_array_ops.cc277 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 Dresource_variable_ops.cc97 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 Dembedding_lookup_sparse_test.cc38 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 Dhashtable_lookup_test.cc37 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 Dlookup_table_op.cc125 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 183 milliseconds