Home
last modified time | relevance | path

Searched defs:result_shape (Results 1 – 25 of 86) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dwhile_test.cc51 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
87 auto result_shape = ShapeUtil::MakeShape(S64, {}); in XLA_TEST_F() local
117 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
150 auto result_shape = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
187 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); in XLA_TEST_F() local
240 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); in XLA_TEST_F() local
299 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); in XLA_TEST_F() local
356 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
409 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
465 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder_test.cc684 const Shape& result_shape = in TEST_F() local
709 const Shape& result_shape = in TEST_F() local
734 const Shape& result_shape = in TEST_F() local
755 const Shape& result_shape = in TEST_F() local
777 const Shape& result_shape = in TEST_F() local
800 const Shape& result_shape = in TEST_F() local
856 const Shape& result_shape = in TEST_F() local
897 const Shape& result_shape = in TEST_F() local
936 const Shape& result_shape = in TEST_F() local
962 const Shape& result_shape = in TEST_F() local
[all …]
H A Dexecutable_build_options.cc190 Shape result_shape(program_shape->result()); in CreateExecutionOptions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc67 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); in TEST_F() local
102 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); in TEST_F() local
186 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F() local
225 Shape result_shape = ShapeUtil::MakeShape(F32, {1, 24}); in TEST_F() local
H A Dcpu_instruction_fusion_test.cc298 Shape result_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
352 Shape result_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
372 Shape result_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
411 Shape result_shape = ShapeUtil::MakeShape(F32, {16}); in TEST_F() local
467 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); in TEST_F() local
H A Ddot_op_emitter_internal.h40 Shape result_shape; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_pad_for_convolutions.cc83 const Shape& result_shape = conv->shape().tuple_shapes(0); in PadConv() local
187 const Shape& result_shape = conv->shape().tuple_shapes(0); in TryResolvePaddedShapesForTensorCore() local
301 const Shape& result_shape = conv->shape().tuple_shapes(0); in TryResolvePaddedShapesForIntegerConvolution() local
H A Dgpu_hlo_cost_analysis.cc90 const Shape& result_shape = [&]() -> const Shape& { in GetConvolutionFlops() local
H A Dgpu_conv_algorithm_picker.cc445 const Shape& result_shape = instr->shape().tuple_shapes(0); in AutotuneOneConvRunner() local
649 const Shape& result_shape = instr->shape().tuple_shapes(0); in PickBestAlgorithmNoCacheCuda() local
H A Dcudnn_support_utils.cc33 const Shape& result_shape = conv.shape().tuple_shapes(0); in CudnnSupportsOptimizedIntegerConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_infeed_test.cc122 const auto result_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
204 const auto result_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dbcast.h96 const Vec& result_shape() const { return result_; } in result_shape() function
378 const Vec& result_shape() const { return result_; } in result_shape() function
H A Dstrided_slice_op.h137 const Vec& result_shape() const { return result_shape_; } in result_shape() function
H A Dtensor_slice_set.cc34 TensorShape result_shape; in Register() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_op.h63 TensorShape result_shape = ExtractShapeFromDescriptor(unranked_descriptor); in ConvertDescriptorToTensor() local
108 TensorShape result_shape = ExtractShapeFromDescriptor(result_desc); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment_test.cc6063 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6101 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6144 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6186 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6231 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6296 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6330 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6368 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6416 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
6472 auto result_shape = ShapeUtil::MakeShape(F32, {kBatch, kOutput}); in TEST_P() local
[all …]
H A Dhlo_cost_analysis.cc322 const Shape& result_shape, in GetDotFlops()
585 const Shape& result_shape = convolution->shape(); in GetConvolutionFlops() local
594 const Shape& result_shape) { in GetConvolutionFlops()
H A Dcomputation_layout.h75 const Shape& result_shape() const { return result_layout_.shape(); } in result_shape() function
H A Dhlo_sharding.cc754 Shape result_shape = shape; in TileShape() local
768 Shape result_shape = shape; in TileShape() local
H A Dhlo_module_util.cc31 const Shape& result_shape) { in ValidateResultShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlegalize_jax_random.cc96 auto result_shape = result_ty.getShape(); in runOnOperation() local
H A Dlegalize_tf.cc705 SmallVector<int64_t, 4> result_shape; in matchAndRewrite() local
853 SmallVector<int64_t, 4> result_shape; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dreduction_benchmark.h37 std::vector<int64_t> result_shape; in ReducedTensorShape() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_lowering_context.h39 const Shape& result_shape() const override { in result_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreshape_util.cc166 Tensor *result_shape = nullptr; in ReshapeSparseTensor() local

1234