| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 51 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 …]
|
| D | outfeed_in_nested_computation_test.cc | 125 Shape result_shape = ShapeUtil::MakeShape(xla::PRED, {}); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 684 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 …]
|
| D | executable_build_options.cc | 190 Shape result_shape(program_shape->result()); in CreateExecutionOptions() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 83 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
|
| D | gpu_hlo_cost_analysis.cc | 90 const Shape& result_shape = [&]() -> const Shape& { in GetConvolutionFlops() local
|
| D | cudnn_support_utils.cc | 33 const Shape& result_shape = conv.shape().tuple_shapes(0); in CudnnSupportsOptimizedIntegerConvolution() local
|
| D | cublas_pad_for_gemms.cc | 36 Shape result_shape = dot->shape(); in PadForGemm() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 67 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
|
| D | dot_op_emitter_internal.h | 40 Shape result_shape; member
|
| D | cpu_instruction_fusion_test.cc | 298 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
|
| D | mlir_emitter.cc | 94 mlir::MLIRContext *context, llvm::IRBuilder<> *b, const Shape &result_shape, in EmitMlirFuncAndCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.h | 75 const Shape& result_shape() const { return result_layout_.shape(); } in result_shape() function
|
| D | hlo_module_util.cc | 31 const Shape& result_shape) { in ValidateResultShape()
|
| D | rng_expander.cc | 78 const Shape& result_shape = rng->shape(); in GetComputationForRng() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 122 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/ |
| D | strided_slice_op.h | 137 const Vec& result_shape() const { return result_shape_; } in result_shape() function
|
| D | bcast.h | 96 const Vec& result_shape() const { return result_; } in result_shape() function
378 const Vec& result_shape() const { return result_; } in result_shape() function
|
| D | tensor_slice_set.cc | 34 TensorShape result_shape; in Register() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_jax_random.cc | 96 auto result_shape = result_ty.getShape(); in runOnOperation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | reduction_benchmark.h | 37 std::vector<int64_t> result_shape; in ReducedTensorShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 166 Tensor *result_shape = nullptr; in ReshapeSparseTensor() local
|
| D | gather_op.cc | 136 TensorShape result_shape; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 88 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 88 xla::Shape result_shape(pshape->result()); in CompileXla() local
|