Home
last modified time | relevance | path

Searched defs:value_shape (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dragged_to_dense_util.cc42 const TensorShapeProto& value_shape, TensorShapeProto* output_shape) { in CombineRaggedTensorToTensorShapes()
97 const TensorShapeProto& value_shape) { in ValidateDefaultValueShape()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfill.h26 void Fill(const RuntimeShape& value_shape, const T* value_data, in Fill()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_placeholder_op_test.py53 def testRaggedPlaceholder(self, dtype, ragged_rank, value_shape, name, argument
Dragged_factory_ops.py318 def placeholder(dtype, ragged_rank, value_shape=None, name=None): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfill_op.cc38 const TensorShape value_shape = ctx->InputShape("value"); in Compile() local
Dtensor_array_ops.cc276 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local
364 const TensorShape value_shape = ctx->InputShape(2); in Compile() local
502 TensorShape value_shape = ctx->InputShape(1); in Compile() local
Dstack_ops.cc216 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dinitializable_lookup_table.h73 TensorShape value_shape() const final { return TensorShape(); } in value_shape() function
Dlookup_table_op.cc145 TensorShape value_shape() const override { return TensorShape(); } in value_shape() function in tensorflow::lookup::MutableHashTableOfScalars
294 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableHashTableOfTensors
558 TensorShape value_shape = MaybeVectorizeShape(value_shape_); in CheckKeyAndValueTensorsForImport() local
580 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableDenseHashTable
/external/tensorflow/tensorflow/core/ops/
Dragged_conversion_ops.cc257 TensorShapeProto value_shape; in RaggedTensorToTensorShapeFn() local
Dresource_variable_ops.cc174 ShapeHandle value_shape = c->input(1); in CreateAssignShapeFn() local
Ddata_flow_ops.cc732 ShapeHandle value_shape = c->input(2); in __anonea01102b1802() local
814 ShapeHandle value_shape; in __anonea01102b1b02() local
/external/tensorflow/tensorflow/lite/kernels/
Dhashtable_lookup_test.cc43 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
Dembedding_lookup_sparse_test.cc42 std::initializer_list<int> value_shape) { in EmbeddingLookupSparseOpModel()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc230 const RuntimeShape& value_shape) { in CombineRaggedTensorToTensorShapes()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc4813 std::initializer_list<int> value_shape, in HashtableLookupOpModel()