| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | resampler_ops.cc | 202 xla::Shape data_shape, XlaOp sample) { in BoundSamples()
255 xla::Shape data_shape) { in CalculateGradData()
365 xla::PrimitiveType data_type, xla::Shape data_shape) { in CalculateGradWarp()
490 TensorShape data_shape = ctx->InputShape("data"); in Compile() local
638 xla::Shape data_shape = TensorShapeToXLAShape(data_type, data_shape_tf); in Compile() local
|
| H A D | reduction_ops_common.cc | 48 const TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
| H A D | dynamic_stitch_op.cc | 75 TensorShape& data_shape = data_shapes[input_num]; in Compile() local
|
| H A D | segment_reduction_ops.cc | 55 TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
| H A D | dynamic_partition_op.cc | 111 xla::Shape data_shape = ctx->InputXlaShape(0).value(); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_dataflow_analysis_test.cc | 2103 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2140 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2211 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2235 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2271 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2297 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2396 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2434 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2474 Shape data_shape = ShapeUtil::MakeShape(F32, {1, 8}); in TEST_F() local
2676 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
[all …]
|
| H A D | rng_bit_generator_expander.cc | 57 const Shape& data_shape, const Shape& state_shape, in GetGeneratorComputation()
110 const Shape& data_shape = rng->shape().tuple_shapes(1); in ExpandInstruction() local
|
| H A D | rng_bit_generator_expander.h | 42 Shape data_shape; member
|
| H A D | while_loop_concat_code_motion.cc | 871 Shape data_shape = hlo->operand(i)->shape(); in RewriteLoopWithConcatGroups() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | math_ops.cc | 1156 ShapeHandle data_shape; in SegmentReductionShapeFn() local
1172 ShapeHandle data_shape; in SparseSegmentReductionShapeFn() local
1196 ShapeHandle data_shape; in SparseSegmentReductionGradShapeFn() local
1234 ShapeHandle data_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
|
| H A D | data_flow_ops.cc | 58 ShapeHandle data_shape = c->input(0); in __anon1926cf7a0202() local
105 ShapeHandle data_shape = c->input(i + num_partitions); in DynamicStitchShapeFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | image_grad_test.cc | 43 Tensor MakeData(const TensorShape& data_shape) { in MakeData()
188 Tensor MakeData(const TensorShape& data_shape) { in MakeData()
305 Tensor MakeData(const TensorShape& data_shape) { in MakeData()
|
| H A D | data_flow_grad_test.cc | 50 TensorShape data_shape({2, 3, 2}); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 42 def test_linear(self, batch_size, data_shape, weight_output_dim, use_bias): argument
250 def test_linear(self, batch_size, data_shape, weight_output_dim, use_bias): argument
768 data_shape, argument
1254 self, pattern_count_transformed_map, pattern_count_optimized_map, data_shape argument
|
| H A D | test_metal.py | 16 data_shape, argument
|
| H A D | test_vulkan.py | 19 data_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/ |
| H A D | tensor_utils.cc | 21 Status QuantizationAxisAndShapeValid(const TensorShape& data_shape, in QuantizationAxisAndShapeValid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | get_item_op.cc | 36 const TensorShape& data_shape = ctx->InputShape(0); in Compile() local
|
| H A D | segment_reduction_ops.cc | 45 TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/client/ |
| H A D | client_session_test.cc | 209 TensorShape data_shape({1, 1}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | tuple_test.cc | 273 Shape data_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | xla.proto | 699 ShapeProto data_shape = 2; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1177 xla::Shape data_shape = xla::ShapeUtil::MakeTupleShape(subshapes); in ExportXlaOp() local
1290 xla::Shape data_shape; in ExportXlaOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 842 const Shape& data_shape = GetShape(pad_to_static.getOutput().front()); in EmitPadToStatic() local
872 const Shape& data_shape = GetShape(slice_to_dynamic.getOutput().front()); in EmitSliceToDynamic() local
2229 const Shape& data_shape = GetShape(scatter_op.getUpdates()); in EmitScatter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 408 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed() local
2323 const Shape& data_shape = ShapeUtil::GetSubshape(hlo->shape(), {0}); in HandlePadToStatic() local
|