| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| tfcompile_test.cc | 565 using xla::ShapeUtil;
566 const xla::Shape f32_2x2 = ShapeUtil::MakeShape(xla::F32, {2, 2});
578 ShapeUtil::Compatible(xla::Shape(muladd_shape->parameters(0)), f32_2x2));
580 ShapeUtil::Compatible(xla::Shape(muladd_shape->parameters(1)), f32_2x2));
584 ASSERT_EQ(ShapeUtil::TupleElementCount(muladd_result), 2);
586 ShapeUtil::GetTupleElementShape(muladd_result, 0);
587 EXPECT_TRUE(ShapeUtil::Compatible(muladd_result0, f32_2x2));
589 ShapeUtil::GetTupleElementShape(muladd_result, 1);
590 EXPECT_TRUE(ShapeUtil::Compatible(muladd_result1, f32_2x2));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_tensor.cc | 59 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first);
|
| xla_launch_util.cc | 245 CHECK(xla::ShapeUtil::Equal(shape, on_device_shape))
247 << xla::ShapeUtil::HumanStringWithLayout(on_device_shape)
249 << xla::ShapeUtil::HumanStringWithLayout(shape);
280 xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_host_shape()}),
281 xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_device_shape()}),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| matrix_band_part_op.cc | 64 xla::Shape iota_shape = xla::ShapeUtil::MakeShape(index_xla_type, {m, n});
|
| tensor_list_utils.cc | 63 xla::ShapeUtil::GetTupleElementShape(list_shape, 0), buffer_shape));
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| comparators.cc | 113 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {});
|
| math_test.cc | 180 shape = ShapeUtil::MakeShape(ty, {42});
182 shape = ShapeUtil::MakeTupleShape({});
184 shape = ShapeUtil::MakeOpaqueShape();
186 shape = ShapeUtil::MakeTokenShape();
|
| prng.cc | 117 const int64 size = ShapeUtil::ElementsIn(shape);
146 const int64 size = ShapeUtil::ElementsIn(shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| bfloat16_normalization.cc | 114 ShapeUtil::ChangeElementType(operand->shape(), to), operand));
162 output_types[i] = ShapeUtil::GetSubshape(hlo->shape(), {i}).element_type();
239 auto subshape = ShapeUtil::GetMutableSubshape(hlo->mutable_shape(), {i});
250 ShapeUtil::ChangeElementType(*subshape, BF16), gte));
|
| heap_simulator_test.cc | 43 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {});
45 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape});
57 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter,
86 return ShapeUtil::ByteSizeOf(buffer.shape(), /*pointer_size=*/8);
131 const Shape r0f32 = ShapeUtil::MakeShape(F32, {});
132 const Shape r1f32 = ShapeUtil::MakeShape(F32, {4});
133 const Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 4});
142 ShapeUtil::MakeShape(F32, {1}), cond_param, {0}, {1}, {1}));
148 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), reshape,
204 return ShapeUtil::ByteSizeOf(buffer.shape())
[all...] |
| shaped_buffer.h | 77 // ShapeUtil::GetSubshape.
91 CHECK(ShapeUtil::Equal(buffers.shape(), on_device_shape_));
|
| while_util.cc | 53 HloInstruction::CreateCall(ShapeUtil::MakeShape(PRED, {}),
154 Shape scalar_pred = ShapeUtil::MakeShape(PRED, {});
220 loop_state_shape_components.push_back(ShapeUtil::MakeShape(S32, {}));
224 return ShapeUtil::MakeTupleShape(loop_state_shape_components);
|
| hlo_creation_utils_test.cc | 34 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims);
36 ShapeUtil::MakeShape(primitive_type, output_shape_dims);
|
| hlo_domain_test.cc | 392 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 0), infeed_data,
399 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 1), infeed_data,
479 Shape shape = ShapeUtil::MakeShape(F32, {});
646 ShapeUtil::GetTupleElementShape(copy1->shape(), 0), copy1, 0));
649 ShapeUtil::GetTupleElementShape(tuple0->shape(), 1), tuple0, 1));
|
| hlo_computation.cc | 280 CHECK(ShapeUtil::Compatible(new_root_instruction->shape(),
490 s << ShapeUtil::HumanString(ComputeProgramShape()) << " ";
631 for (int64 i = 0; i < ShapeUtil::TupleElementCount(instruction->shape());
635 ShapeUtil::GetTupleElementShape(instruction->shape(), i),
666 !ShapeUtil::Compatible(instruction->shape(), indices_to_copy->shape())) {
670 instruction->name(), ShapeUtil::HumanString(instruction->shape()),
671 ShapeUtil::HumanString(indices_to_copy->shape()));
765 ShapeUtil::Compatible(old_instruction->shape(), new_instruction->shape()))
766 << ShapeUtil::HumanString(old_instruction->shape()) << " vs "
767 << ShapeUtil::HumanString(new_instruction->shape())
[all...] |
| hlo_instruction.cc | 134 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape));
360 (ShapeUtil::TupleElementCount(shape) == 2))
363 const Shape& data_shape = ShapeUtil::GetTupleElementShape(shape, 0);
370 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_intrinsic_test.cc | 80 auto param_shape = ShapeUtil::MakeShape(F32, {1024});
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cholesky_thunk.cc | 49 ShapeUtil::ByteSizeOfPrimitiveType(
|
| cudnn_conv_algorithm_picker.cc | 215 &stream, ShapeUtil::ByteSizeOf(operand->shape())));
222 &stream, ShapeUtil::ByteSizeOf(instr->shape().tuple_shapes(0))));
379 Shape new_call_shape = ShapeUtil::MakeTupleShape(
381 ShapeUtil::MakeShape(U8, {best_algo.success().scratch_bytes()})});
|
| cudnn_conv_rewriter.cc | 56 ShapeUtil::MakeTupleShape({shape, ShapeUtil::MakeShape(U8, {0})});
75 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) ||
76 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) {
|
| nvptx_compiler.h | 71 return ShapeUtil::ByteSizeOf(shape, pointer_size);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_dimension_numbers_test.cc | 102 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {4, 3, 1, 1}), "weight");
|
| batch_normalization_test.cc | 76 CHECK(ShapeUtil::Equal(expected_shape, actual_shape))
77 << "want " << ShapeUtil::HumanString(expected_shape) << " got "
78 << ShapeUtil::HumanString(actual_shape);
176 ShapeUtil::MakeShape(F32, {3, 2, 1, 1}));
181 Shape TwoElementVectorF32 = ShapeUtil::MakeShape(F32, {2});
190 ConstantR0<float>(&builder, ShapeUtil::ElementsIn(input_shape) /
191 ShapeUtil::ElementsIn(sum_shape));
209 ShapeUtil::MakeShape(PRED, {2}));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| text_literal_reader.cc | 74 ShapeUtil::HumanString(shape));
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| dynamic_update_slice_util.cc | 127 bool is_signed = ShapeUtil::ElementIsSigned(start_indices_array.GetShape());
179 bool is_signed = ShapeUtil::ElementIsSigned(start_indices->shape());
|
