| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | ragged_to_dense_util.cc | 42 const TensorShapeProto& value_shape, TensorShapeProto* output_shape) { in CombineRaggedTensorToTensorShapes()
97 const TensorShapeProto& value_shape) { in ValidateDefaultValueShape()
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | flop_counter.py | 241 def sdpa_flop_count(query_shape, key_shape, value_shape): argument
260 def sdpa_flop(query_shape, key_shape, value_shape, *args, out_shape=None, **kwargs) -> int: argument
436 def sdpa_backward_flop_count(grad_out_shape, query_shape, key_shape, value_shape): argument
464 def sdpa_backward_flop(grad_out_shape, query_shape, key_shape, value_shape, *args, out_shape=None, … argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_array_ops.cc | 276 std::vector<int64_t> value_shape(slice_shape.begin() + 1, in Compile() local
365 const TensorShape value_shape = ctx->InputShape(2); in Compile() local
503 TensorShape value_shape = ctx->InputShape(1); in Compile() local
|
| H A D | fill_op.cc | 39 const TensorShape value_shape = ctx->InputShape("value"); in Compile() local
|
| H A D | stack_ops.cc | 216 std::vector<int64_t> value_shape(slice_shape.begin() + 1, in Compile() local
|
| H A D | variable_ops.cc | 36 TensorShape value_shape = ctx->InputShape(1); in ValidateAssignUpdateVariableOpShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | lookup_table_op.cc | 144 TensorShape value_shape() const override { return TensorShape(); } in value_shape() function in tensorflow::lookup::MutableHashTableOfScalars
332 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableHashTableOfTensors
645 TensorShape value_shape = MaybeVectorizeShape(value_shape_); in CheckKeyAndValueTensorsForImport() local
667 TensorShape value_shape() const override { return value_shape_; } in value_shape() function in tensorflow::lookup::MutableDenseHashTable
|
| H A D | resource_variable_util.cc | 23 const TensorShape& value_shape) { in ValidateAssignUpdateVariableOpShapes()
|
| H A D | initializable_lookup_table.h | 74 TensorShape value_shape() const final { return TensorShape(); } in value_shape() function
|
| H A D | ragged_tensor_from_variant_op.cc | 240 TensorShape value_shape = ragged_components[i].values().shape(); in NestedStackRaggedTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | fill.h | 26 void Fill(const RuntimeShape& value_shape, const T* value_data, in Fill()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_placeholder_op_test.py | 49 def testRaggedPlaceholder(self, dtype, ragged_rank, value_shape, name, argument
|
| H A D | ragged_factory_ops.py | 320 def placeholder(dtype, ragged_rank, value_shape=None, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | data_flow_ops.cc | 732 ShapeHandle value_shape = c->input(2); in __anon1926cf7a1802() local
814 ShapeHandle value_shape; in __anon1926cf7a1b02() local
|
| H A D | ragged_conversion_ops.cc | 254 TensorShapeProto value_shape; in RaggedTensorToTensorShapeFn() local
|
| H A D | resource_variable_ops.cc | 189 ShapeHandle value_shape = c->input(1); in CreateAssignShapeFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_lookup_test.cc | 43 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
|
| H A D | embedding_lookup_sparse_test.cc | 42 std::initializer_list<int> value_shape) { in EmbeddingLookupSparseOpModel()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HashtableLookupTest.cpp | 63 std::initializer_list<uint32_t> value_shape) { in HashtableLookupOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | restore.py | 364 def value_shape(self): member in CheckpointPosition
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_tensor_to_tensor_tflite.cc | 230 const RuntimeShape& value_shape) { in CombineRaggedTensorToTensorShapes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy.py | 1715 def _compute_all_gather_output_shape(self, value_shape, value_rank, axis): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 4890 std::initializer_list<int> value_shape, in HashtableLookupOpModel()
|