| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | weights_broadcast_ops.py | 30 def _has_valid_dims(weights_shape, values_shape): argument
44 weights_rank, weights_shape, values_rank, values_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing.cc | 1280 TensorShape values_shape; in FastParseExample() local
1338 TensorShape values_shape; in FastParseExample() local
1391 TensorShape values_shape; in FastParseExample() local
1498 TensorShape values_shape; in FastParseSingleExample() local
2366 TensorShape indices_shape, values_shape; in ParseContextSparseFeatures() local
2428 TensorShape values_shape, splits_shape; in ParseContextRaggedFeatures() local
2659 TensorShape indices_shape, values_shape; in ParseSequenceSparseFeatures() local
2787 TensorShape values_shape, inner_splits_shape, outer_splits_shape; in ParseSequenceRaggedFeatures() local
|
| H A D | example_proto_helper.h | 88 TensorShape values_shape; member
|
| H A D | example_proto_helper.cc | 382 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_variant_op.cc | 41 TensorShape values_shape = batched_values.shape(); in UnbatchDenseZerothDim() local
92 TensorShape values_shape = batched_ragged.values().shape(); in UnbatchRaggedZerothDim() local
|
| H A D | ragged_tensor_to_sparse_kernel.cc | 202 const TensorShape& values_shape) { in MakeIndexSuffixes()
|
| H A D | example_parsing_ops.cc | 1052 TensorShape values_shape({0}); in Compute() local
1162 TensorShape values_shape({total_num_features}); in Compute() local
|
| H A D | ragged_gather_op.cc | 246 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorTransformerFunctions.cpp | 267 auto values_shape = values.sizes().vec(); in _jagged_to_padded_dense_forward_cpu() local
320 std::vector<int64_t> values_shape; in _padded_dense_to_jagged_forward_cpu() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | lower_upper_bound_ops.cc | 36 const TensorShape values_shape = ctx->InputShape("values"); in BuildLowerUpperBoundOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | sparse_to_dense_test.cc | 40 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_tensor_to_tensor_tflite_test.cc | 42 std::initializer_list<int> values_shape, in RaggedTensorToTensorOpModel()
|
| H A D | ragged_tensor_to_tensor_tflite.cc | 254 RuntimeShape values_shape(values.dims->size, values.dims->data); in CalculateOutputSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 665 auto values_shape = Shape(scope, values); in BatchGatherGrad() local
730 auto values_shape = Concat(scope, {outer_shape, {-1}, inner_shape}, 0); in GatherV2Grad() local
|
| /aosp_15_r20/external/executorch/backends/apple/mps/serialization/ |
| H A D | mps_graph_schema.py | 473 values_shape: List[int] = field(default_factory=list) variable in MPSIndexPut
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 555 tf::TensorShape values_shape; in FastParseExampleLite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_cost_analysis_test.cc | 1175 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
|
| H A D | hlo_dataflow_analysis_test.cc | 2647 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| H A D | algebraic_simplifier_test.cc | 4017 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 1778 DimVector values_shape{num_blocks, blocksize[0], blocksize[1]}; in _compressed_to_block_compressed_cpu() local
|
| H A D | BatchLinearAlgebra.cpp | 2880 auto values_shape = IntArrayRef(input.sizes().data(), input.dim()-1); // input.shape[:-1] in linalg_eig_out_info() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 2319 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
|