| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fft_thunk.cc | 113 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_);
115 << ShapeUtil::HumanStringWithLayout(output_shape_);
178 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_),
202 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_),
|
| fusion_merger.cc | 77 bytes += ShapeUtil::ByteSizeOf(user->shape());
81 bytes += ShapeUtil::ByteSizeOf(instruction->shape());
107 bytes_written += ShapeUtil::ByteSizeOf(operand->shape());
111 ShapeUtil::ByteSizeOf(fusion->fused_expression_root()->shape());
|
| ir_emission_utils.cc | 61 !ShapeUtil::IsZeroElementArray(lhs_shape) &&
62 !ShapeUtil::IsZeroElementArray(rhs_shape);
173 ShapeUtil::Equal(
175 ShapeUtil::FilterDimensions(
|
| partition_assignment.cc | 65 int64 num_elements = ShapeUtil::ElementsIn(shape);
|
| ir_emitter.cc | 202 store->setAlignment(ShapeUtil::ByteSizeOfPrimitiveType(element_type));
432 TF_RET_CHECK(ShapeUtil::IsScalar(pred->shape()));
483 if (ShapeUtil::IsScalar(lhs_shape) && ShapeUtil::IsScalar(rhs_shape)) {
490 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
495 } else if (ShapeUtil::ElementIsFloating(lhs_shape)) {
498 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape));
507 TF_RET_CHECK(!ShapeUtil::IsScalar(lhs_shape) &&
508 !ShapeUtil::IsScalar(rhs_shape));
587 if (ShapeUtil::ElementIsComplex(lhs_shape))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_module_dce.cc | 62 ShapeUtil::TupleElementCount(xla_while->shape());
|
| local_service.cc | 137 ShapeUtil::ValidateShapeWithOptionalLayout(argument_shape));
138 if (!ShapeUtil::Compatible(argument_shape, program_shape.parameters(i))) {
155 ShapeUtil::HumanString(program_shape.parameters(i)),
156 ShapeUtil::HumanString(argument_shape));
|
| map_inliner_test.cc | 42 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
79 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
112 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
150 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
|
| stable_sort_expander.cc | 84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {});
105 Shape new_sort_shape = ShapeUtil::MakeTupleShape(new_shapes);
181 Shape scalar_pred = ShapeUtil::MakeShape(PRED, {});
192 ShapeUtil::MakeShape(PRED, {}), HloOpcode::kSelect, same, tie_breaker,
|
| while_loop_invariant_code_motion.cc | 221 ShapeUtil::ForEachSubshape(
225 input_size += ShapeUtil::ByteSizeOfElements(subshape);
229 ShapeUtil::ForEachSubshape(
233 output_size += ShapeUtil::ByteSizeOfElements(subshape);
|
| flatten_call_graph_test.cc | 85 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param0,
96 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {});
159 0, ShapeUtil::MakeShape(PRED, {}), "param0"));
163 ShapeUtil::MakeShape(PRED, {}), param0, false_constant,
174 ShapeUtil::MakeShape(PRED, {}), cond_computation, cond_computation,
|
| hlo_evaluator.cc | 246 if (!ShapeUtil::Equal(computation_shape, arg_shape)) {
250 i, ShapeUtil::HumanStringWithLayout(computation_shape),
251 ShapeUtil::HumanString(arg_shape));
410 Literal output(ShapeUtil::MakeShape(U32, {}));
425 DCHECK(ShapeUtil::Equal(parameter->shape(), input_literal->shape()))
426 << "parameter shape is: " << ShapeUtil::HumanString(parameter->shape())
428 << ShapeUtil::HumanString(input_literal->shape());
470 ShapeUtil::GetDimension(operand_shape, concat_dim);
484 ShapeUtil::GetDimension(operand_shape, concat_dim);
643 TF_RET_CHECK(ShapeUtil::Compatible(real.shape(), imag.shape()))
[all...] |
| hlo_rematerialization_test.cc | 70 ShapeUtil::MakeShape(xla::F32, {2048}), {negate, negate},
77 ShapeUtil::MakeShape(xla::F32, {1025}), {bcast, slice_1},
119 ShapeUtil::MakeShape(xla::F32, {1025}), {bcast, while_inst},
141 return ShapeUtil::ByteSizeOf(shape, sizeof(void*));
157 const Shape scalar_shape_ = ShapeUtil::MakeShape(xla::F32, {});
158 const Shape vec1_shape_ = ShapeUtil::MakeShape(xla::F32, {1});
159 const Shape vec1024_shape_ = ShapeUtil::MakeShape(xla::F32, {1024});
431 ShapeUtil::MakeShape(xla::F32, {2048}), {param, param},
590 ShapeUtil::MakeShape(xla::F32, {2048}), {param, param},
|
| buffer_assignment_test.cc | 194 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), index,
299 Shape s32_ = ShapeUtil::MakeShape(xla::S32, {});
300 Shape r0f32_ = ShapeUtil::MakeShape(xla::F32, {});
301 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4});
302 Shape f32vec10_ = ShapeUtil::MakeShape(F32, {10});
303 Shape f32vec100_ = ShapeUtil::MakeShape(F32, {100});
304 Shape f32a100x10_ = ShapeUtil::MakeShape(F32, {100, 10});
305 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
306 Shape t_s32_f32v10_ = ShapeUtil::MakeTupleShape({s32_, f32vec10_});
[all...] |
| indexed_array_analysis.cc | 50 return absl::StrCat("(constant ", ShapeUtil::HumanString(root->shape()),
53 return absl::StrCat("(constant ", ShapeUtil::HumanString(root->shape()),
61 " to ", ShapeUtil::HumanString(reshaped_array->shape()), ")");
246 ShapeUtil::MakeShape(x->shape().element_type(),
461 return ShapeUtil::MakeShape(shape.element_type(), new_dims);
469 if (!ShapeUtil::HasDegenerateDimensions(shape)) {
485 ShapeUtil::MakeShape(shape.element_type(), new_source_shape_dims);
528 CHECK(!ShapeUtil::HasDegenerateDimensions(operand->shape()));
586 Shape new_source_shape = ShapeUtil::MakeShape(operand->shape().element_type(),
588 Shape new_result_shape = ShapeUtil::MakeShape(operand->shape().element_type()
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| axpy_simple_test.cc | 67 EXPECT_EQ("() -> f32[10]", ShapeUtil::HumanString(shape));
|
| execution_profile_test.cc | 29 Shape shape = ShapeUtil::MakeShape(F32, {256, 256});
|
| xla_hlo_profile_test.cc | 192 Shape lhs_shape = ShapeUtil::MakeShape(F32, {m, k});
193 Shape rhs_shape = ShapeUtil::MakeShape(F32, {m, k});
202 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {m, k}), "dot_lhs"),
203 Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {k, n}), "dot_rhs")));
270 Shape matrix_shape = ShapeUtil::MakeShape(F32, {size, size});
272 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(S32, {}), matrix_shape});
|
| params_test.cc | 49 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "param0");
61 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {0}), "param0");
73 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {2}), "param0");
87 ShapeUtil::MakeShape(U8, {static_cast<int64>(str.size())}),
100 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {3, 0}), "param0");
113 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {3, 2}), "param0");
160 Parameter(&builder, 2, ShapeUtil::MakeShape(F32, {}), "param2");
378 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes);
436 Shape r1f32_3 = ShapeUtil::MakeShape(F32, {3});
437 Shape tuple_shape = ShapeUtil::MakeTupleShape({r1f32_3, r1f32_3})
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util.cc | 191 ShapeUtil::HumanString(shape));
382 if (ShapeUtil::TupleElementCount(src) !=
383 ShapeUtil::TupleElementCount(*dst)) {
387 for (int64 i = 0; i < ShapeUtil::TupleElementCount(src); ++i) {
416 if (!rhs.IsTuple() || ShapeUtil::TupleElementCount(lhs) !=
417 ShapeUtil::TupleElementCount(rhs)) {
420 for (int i = 0; i < ShapeUtil::TupleElementCount(lhs); ++i) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| kernel_tiling.cc | 52 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(),
82 if (!ShapeUtil::CompatibleIgnoringElementType(a, b)) {
100 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(a);
153 Shape output_shape = ShapeUtil::MakeShapeWithDescendingLayout(
167 ShapeUtil::MakeShapeWithDescendingLayout(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.cc | 91 if (!ShapeUtil::ElementIsIntegral(shape)) {
94 ShapeUtil::HumanString(shape));
96 if (ShapeUtil::ElementIsSigned(shape)) {
405 if (!ShapeUtil::IndexIsValid(parameter_shape, alias.param_index)) {
437 CHECK(ShapeUtil::IsScalar(operand_shape) ||
440 ShapeUtil::ChangeElementType(output_shape, operand_shape.element_type());
443 if (ShapeUtil::IsScalar(operand_shape)) {
463 Reshape(ShapeUtil::MakeShape(operand_shape.element_type(),
528 const Shape& broadcasted_shape = ShapeUtil::MakeShape(
539 if (!ShapeUtil::SameDimensions(shape, updated_lhs_shape))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| numpy_bridge.cc | 136 int num_elements = ShapeUtil::TupleElementCount(shape);
137 dimensions = make_safe(PyTuple_New(ShapeUtil::TupleElementCount(shape)));
141 PyShapeInfoFromXlaShape(ShapeUtil::GetTupleElementShape(shape, i))
149 LongToPyIntOrPyLong(ShapeUtil::GetDimension(shape, i)));
256 return ShapeUtil::MakeTupleShape(subshapes);
297 return ShapeUtil::MakeShapeWithLayout(element_type, dimensions,
300 return ShapeUtil::MakeShape(element_type, dimensions);
372 int num_elements = ShapeUtil::TupleElementCount(literal.shape());
387 dimensions[i] = ShapeUtil::GetDimension(literal.shape(), i);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 236 int64 byte_size = ShapeUtil::ByteSizeOfPrimitiveType(primitive_type);
252 if (ShapeUtil::IsScalar(shape)) {
316 TF_RET_CHECK(ShapeUtil::IsScalar(pred->shape()));
332 DCHECK(ShapeUtil::Equal(data_shape,
333 ShapeUtil::GetTupleElementShape(infeed->shape(), 0)));
343 token_slice, ShapeUtil::GetTupleElementShape(infeed->shape(), 1));
347 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape));
359 ShapeUtil::GetTupleElementShape(data_shape, i);
489 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape));
493 ShapeUtil::GetTupleElementShape(operand_shape, i)
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| categorical_op.cc | 73 xla::ShapeUtil::MakeShape(uniform_xla_type, uniform_shape_array);
85 xla::ShapeUtil::MakeShape(uniform_xla_type, uniform_shape_array);
|
