/external/tensorflow/tensorflow/core/util/ |
D | ragged_to_dense_util.cc | 42 const TensorShapeProto& value_shape, TensorShapeProto* output_shape) { in CombineRaggedTensorToTensorShapes() 97 const TensorShapeProto& value_shape) { in ValidateDefaultValueShape()
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | fill.h | 26 void Fill(const RuntimeShape& value_shape, const T* value_data, in Fill()
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_placeholder_op_test.py | 53 def testRaggedPlaceholder(self, dtype, ragged_rank, value_shape, name, argument
|
D | ragged_factory_ops.py | 318 def placeholder(dtype, ragged_rank, value_shape=None, name=None): argument
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | fill_op.cc | 38 const TensorShape value_shape = ctx->InputShape("value"); in Compile() local
|
D | tensor_array_ops.cc | 276 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
|
D | stack_ops.cc | 216 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end()); in Compile() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | initializable_lookup_table.h | 73 TensorShape value_shape() const final { return TensorShape(); } in value_shape() function
|
D | lookup_table_op.cc | 145 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/ |
D | ragged_conversion_ops.cc | 257 TensorShapeProto value_shape; in RaggedTensorToTensorShapeFn() local
|
D | resource_variable_ops.cc | 174 ShapeHandle value_shape = c->input(1); in CreateAssignShapeFn() local
|
D | data_flow_ops.cc | 732 ShapeHandle value_shape = c->input(2); in __anonea01102b1802() local 814 ShapeHandle value_shape; in __anonea01102b1b02() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | hashtable_lookup_test.cc | 43 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
|
D | embedding_lookup_sparse_test.cc | 42 std::initializer_list<int> value_shape) { in EmbeddingLookupSparseOpModel()
|
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
D | ragged_tensor_to_tensor_tflite.cc | 230 const RuntimeShape& value_shape) { in CombineRaggedTensorToTensorShapes()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 4813 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
|