Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc52 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
88 auto result_shape = ShapeUtil::MakeShape(S64, {}); in XLA_TEST_F() local
118 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
151 auto result_shape = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F() local
188 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); in XLA_TEST_F() local
242 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); in XLA_TEST_F() local
302 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); in XLA_TEST_F() local
360 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
414 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
471 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F() local
[all …]
Doutfeed_in_nested_computation_test.cc125 Shape result_shape = ShapeUtil::MakeShape(xla::PRED, {}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc614 const Shape& result_shape = in TEST_F() local
639 const Shape& result_shape = in TEST_F() local
664 const Shape& result_shape = in TEST_F() local
685 const Shape& result_shape = in TEST_F() local
707 const Shape& result_shape = in TEST_F() local
730 const Shape& result_shape = in TEST_F() local
758 const Shape& result_shape = in TEST_F() local
799 const Shape& result_shape = in TEST_F() local
838 const Shape& result_shape = in TEST_F() local
864 const Shape& result_shape = in TEST_F() local
[all …]
Dexecutable_build_options.cc120 Shape result_shape(program_shape->result()); in CreateExecutionOptions() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc86 const Shape& result_shape = conv->shape().tuple_shapes(0); in PadConv() local
185 const Shape& result_shape = conv->shape().tuple_shapes(0); in TryResolvePaddedShapesForTensorCore() local
286 const Shape& result_shape = conv->shape().tuple_shapes(0); in TryResolvePaddedShapesForIntegerConvolution() local
Dgpu_layout_assignment_test.cc125 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
199 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
272 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
Dcublas_gemm_pad_for_tensor_cores.cc35 Shape result_shape = dot->shape(); in PadForTensorCores() local
Dgpu_layout_assignment.cc125 Shape result_shape = instr->shape().tuple_shapes(0); in AddBackendConstraintsToDnnConvCustomCall() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); in TEST_F() local
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {24}, {0}); in TEST_F() local
187 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F() local
226 Shape result_shape = ShapeUtil::MakeShape(F32, {1, 24}); in TEST_F() local
Ddot_op_emitter_internal.h40 Shape result_shape; member
Dcpu_instruction_fusion_test.cc316 Shape result_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
370 Shape result_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
390 Shape result_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
429 Shape result_shape = ShapeUtil::MakeShape(F32, {16}); in TEST_F() local
485 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); in TEST_F() local
Dmlir_emitter.cc91 mlir::MLIRContext *context, llvm::IRBuilder<> *b, const Shape &result_shape, in EmitMlirFuncAndCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h75 const Shape& result_shape() const { return result_layout_.shape(); } in result_shape() function
Dhlo_module_util.cc31 const Shape& result_shape) { in ValidateResultShape()
Drng_expander.cc76 const Shape& result_shape = rng->shape(); in GetComputationForRng() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
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
/external/tensorflow/tensorflow/core/util/
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
Dtensor_slice_set.cc34 TensorShape result_shape; in Register() local
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_op.h86 TensorShape result_shape = ExtractShapeFromDescriptor(unranked_descriptor); in ConvertDescriptorToTensor() local
127 TensorShape result_shape = ExtractShapeFromDescriptor(result_desc); in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.cc160 Tensor *result_shape = nullptr; in ReshapeSparseTensor() local
Dgather_op.cc136 TensorShape result_shape; in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfold_broadcast.cc86 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
/external/tensorflow/tensorflow/compiler/aot/
Dcompile.cc88 xla::Shape result_shape(pshape->result()); in CompileXla() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc78 Shape result_shape; in GetShardedProgramShapes() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc118 std::vector<int64> result_shape(result_rank); in Compile() local

123