| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| concat_test.cc | 411 auto opaque_shape = ShapeUtil::MakeOpaqueShape();
412 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1});
426 auto token_shape = ShapeUtil::MakeTokenShape();
427 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1});
562 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {});
584 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {});
611 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {});
|
| vector_ops_simple_test.cc | 328 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x_value");
341 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "y_value");
354 Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "z_value");
|
| cholesky_test.cc | 167 Shape shape = ShapeUtil::MakeShape(F32, dimensions);
|
| convert_test.cc | 395 auto param = Parameter(b.get(), 0, ShapeUtil::MakeShape(F32, {}), "in");
407 auto param = Parameter(b.get(), 0, ShapeUtil::MakeShape(S32, {}), "in");
462 ShapeUtil::MakeShape(F16, {static_cast<int64>(input.size())}),
482 ShapeUtil::MakeShape(F32, {static_cast<int64>(input.size())}),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_alias_analysis_test.cc | 114 const Shape scalar_shape_ = ShapeUtil::MakeShape(F32, {});
223 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
275 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
336 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
524 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
619 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
692 ShapeUtil::MakeTupleShape({scalar_shape_, scalar_shape_});
[all...] |
| ar_crs_combiner.cc | 55 return ShapeUtil::ElementIsFloating(instruction->shape()) ==
56 ShapeUtil::ElementIsFloating(instruction->operand(0)->shape());
61 return ShapeUtil::ElementIsFloating(instruction->shape());
|
| while_loop_analysis.cc | 171 if (!ShapeUtil::IsEffectiveScalar(l.shape())) {
479 CHECK(ShapeUtil::Equal(cond_result_pred.shape(),
480 ShapeUtil::MakeShape(PRED, {})));
|
| pattern_matcher.h | 691 if (!ShapeUtil::Equal(*shape_, *shape)) {
693 << ShapeUtil::HumanStringWithLayout(*shape_);
700 *os << "equal to " << ShapeUtil::HumanStringWithLayout(*shape_);
715 if (!ShapeUtil::Compatible(*shape_, *shape)) {
717 << ShapeUtil::HumanString(*shape_);
724 *os << "compatible with " << ShapeUtil::HumanString(*shape_);
761 if (!ShapeUtil::IsScalar(*shape)) {
816 if (!ShapeUtil::IsEffectiveScalar(*shape)) {
920 return ShapeUtil::GetMutableSubshape(shape, index_);
923 return &ShapeUtil::GetSubshape(*shape, index_)
[all...] |
| logical_buffer_analysis.cc | 102 ShapeUtil::ForEachSubshape(
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xrt.cc | 262 ShapeUtil::HumanString(tuple_shape));
274 for (int32 i = 0; i < ShapeUtil::TupleElementCount(tuple_shape); ++i) {
289 ShapeUtil::GetTupleElementShape(tuple_shape, i);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_cpu.cc | 106 xla::Shape xla_shape = xla::ShapeUtil::MakeShapeWithDescendingLayout(
|
| random_ops.cc | 127 xla::ShapeUtil::MakeShape(xla::U32, {num_elements});
149 auto swaps_shape = xla::ShapeUtil::MakeShape(xla::S32, {n});
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_expression_test.cc | 43 builder_.get(), 0, xla::ShapeUtil::MakeShape(xla::F32, {}), "x");
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| constants.h | 103 if (ShapeUtil::IsScalar(shape) || shape.IsArray()) {
|
| constants_test.cc | 79 Zeros(&builder, ShapeUtil::MakeShape(S32, {2, 2}));
|
| loops.cc | 37 Shape tuple_shape = ShapeUtil::MakeTupleShape(var_shapes);
|
| svd_test.cc | 73 int64 m = ShapeUtil::GetDimension(u_shape, -1);
74 int64 n = ShapeUtil::GetDimension(v_shape, -1);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_runtime.cc | 100 return xla::ShapeUtil::HumanStringWithLayout(shape.ValueOrDie());
|
| vector_support_library.cc | 215 pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
223 pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
233 ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
243 ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| buffer_allocations.cc | 179 return ShapeUtil::IsScalar(literal.shape());
|
| gpu_hlo_schedule.cc | 206 return ShapeUtil::ByteSizeOf(buffer.shape(), pointer_size);
|
| parallel_loop_emitter.cc | 135 llvm::ConstantInt::get(index_type, ShapeUtil::ElementsIn(shape_))),
|
| stream_assignment_test.cc | 42 Shape f32_2x2_ = ShapeUtil::MakeShape(F32, {2, 2});
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| alias_analysis.cc | 143 ShapeUtil::ForEachSubshape(
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| interactive_graphviz.cc | 269 << ShapeUtil::HumanStringWithLayout(param->shape()) << ")"
274 << ShapeUtil::HumanStringWithLayout(root->shape()) << ")"
302 std::cout << " Shape: " << ShapeUtil::HumanStringWithLayout(instr->shape())
326 << ShapeUtil::HumanStringWithLayout(operand->shape()) << ")"
|
