Home
last modified time | relevance | path

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

12

/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.cc516 const Shape& result_shape = in TEST_F() local
541 const Shape& result_shape = in TEST_F() local
566 const Shape& result_shape = in TEST_F() local
587 const Shape& result_shape = in TEST_F() local
609 const Shape& result_shape = in TEST_F() local
632 const Shape& result_shape = in TEST_F() local
660 const Shape& result_shape = in TEST_F() local
701 const Shape& result_shape = in TEST_F() local
740 const Shape& result_shape = in TEST_F() local
766 const Shape& result_shape = in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc123 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
197 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
270 for (const Shape& result_shape : AllLayoutsOf(shape)) { in TEST_F() local
Dcudnn_conv_pad_for_tensor_cores.cc85 const Shape& result_shape = conv->shape().tuple_shapes(0); in PadConv() local
126 const Shape& result_shape = conv->shape().tuple_shapes(0); in PadForTensorCores() local
Dgpu_layout_assignment.cc103 Shape result_shape = instr->shape().tuple_shapes(0); in AddBackendConstraintsToDnnConvCustomCall() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h75 const Shape& result_shape() const { return result_layout_.shape(); } in result_shape() function
Dexecutable.h173 const Shape& result_shape() const { in result_shape() function
Dlocal_service.cc108 Shape result_shape(program_shape->result()); in CreateExecutionOptions() local
Dhlo_module.h115 const Shape& result_shape() const { in result_shape() function
Dhlo_evaluator_typed_visitor.h980 const auto result_shape = reverse->shape(); in HandleReverse() local
1013 const Shape& result_shape = conv->shape(); in HandleConvolution() local
1459 auto result_shape = dynamic_slice->shape(); in HandleDynamicSlice() local
1522 auto result_shape = dynamic_update_slice->shape(); in HandleDynamicUpdateSlice() local
1718 const Shape& result_shape = reduce->shape().IsTuple() in HandleReduce() local
2701 const auto result_shape = random->shape(); in HandleRng() local
2761 const auto result_shape = random->shape(); in HandleRng() local
2861 const Shape& result_shape) { in DynamicSlice()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter_internal.h40 Shape result_shape; member
Dcpu_layout_assignment_test.cc68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); in TEST_F() local
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1}); 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
Dcpu_instruction_fusion_test.cc314 Shape result_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
368 Shape result_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
388 Shape result_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
427 Shape result_shape = ShapeUtil::MakeShape(F32, {16}); in TEST_F() local
483 Shape result_shape = ShapeUtil::MakeShape(F32, {4, 3}); in TEST_F() local
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
Darray_ops.cc55 tensorflow::TensorShape result_shape; in __anon96b295990102() local
/external/tensorflow/tensorflow/core/util/
Dbcast.h99 const Vec& result_shape() const { return result_; } in result_shape() function
Dtensor_slice_set.cc35 TensorShape result_shape; in Register() local
/external/tensorflow/tensorflow/compiler/aot/
Dcompile.cc74 xla::Shape result_shape(pshape->result()); in CompileXla() 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/kernels/
Dreshape_util.cc143 Tensor *result_shape = nullptr; in Reshape() local
Dgather_op.cc89 TensorShape result_shape; in Compute() local
Dgather_nd_op.cc126 TensorShape result_shape(indices_shape); in DoGatherNd() local
Ddynamic_stitch_op.cc127 TensorShape result_shape; in CheckArgsAndAllocateResult() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc118 std::vector<int64> result_shape(result_rank); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc49 Shape result_shape(literal.shape()); in ConvertType() local

12