| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dynamic_parameter_binding.cc | 118 TF_RET_CHECK(ShapeUtil::IndexIsValid(
121 TF_RET_CHECK(ShapeUtil::IndexIsValid(
126 ShapeUtil::GetSubshape(
|
| transpose_folding_test.cc | 222 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {2, 3, 1, 1}),
225 /*parameter_number=*/1, ShapeUtil::MakeShape(F32, {3, 2, 1, 1}),
229 ShapeUtil::MakeShape(F32, {2, 3, 1, 1}), y, {1, 0, 2, 3}));
278 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {2, 3, 1, 1}),
281 /*parameter_number=*/1, ShapeUtil::MakeShape(F32, {1, 2, 1, 3}),
285 ShapeUtil::MakeShape(F32, {2, 3, 1, 1}), y, {1, 3, 0, 2}));
340 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {3, 2, 1, 1}),
343 /*parameter_number=*/1, ShapeUtil::MakeShape(F32, {2, 3, 1, 1}),
347 ShapeUtil::MakeShape(F32, {2, 3, 1, 1}), x, {1, 0, 2, 3}));
407 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {3, 2, 1, 1})
[all...] |
| tuple_simplifier_test.cc | 52 const Shape scalar_shape_ = ShapeUtil::MakeShape(F32, {});
53 const Shape tuple_shape_ = ShapeUtil::MakeTupleShape(
54 {ShapeUtil::MakeShape(F32, {}), ShapeUtil::MakeShape(F32, {}),
55 ShapeUtil::MakeShape(F32, {})});
157 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0));
|
| hlo_casting_utils_test.cc | 27 : HloInstruction(HloOpcode::kConstant, ShapeUtil::MakeShape(F32, {})) {}
33 : HloInstruction(HloOpcode::kParameter, ShapeUtil::MakeShape(F32, {})) {}
|
| tuple_points_to_analysis_test.cc | 363 ShapeUtil::MakeShape(F32, {1, 2, 3}), token, /*channel_id=*/0));
412 Shape tuple_shape = ShapeUtil::MakeTupleShape(
413 {ShapeUtil::MakeShape(F32, {1, 2, 3}), ShapeUtil::MakeShape(U32, {5})});
625 Shape input_shape = ShapeUtil::MakeShape(F32, {8});
626 Shape update_shape = ShapeUtil::MakeShape(F32, {3});
627 Shape starts_shape = ShapeUtil::MakeShape(S32, {});
629 ShapeUtil::MakeTupleShape({input_shape, update_shape, starts_shape});
[all...] |
| reduce_precision_insertion_test.cc | 49 Shape shape = ShapeUtil::MakeShape(F32, {4});
76 Shape shape = ShapeUtil::MakeShape(F32, {});
103 Shape shape = ShapeUtil::MakeShape(F32, {4});
135 Shape shape = ShapeUtil::MakeShape(F32, {4});
163 Shape shape = ShapeUtil::MakeShape(F32, {4});
200 Shape shape = ShapeUtil::MakeShape(F32, {4});
225 Shape shape = ShapeUtil::MakeShape(F32, {4});
266 Shape shape = ShapeUtil::MakeShape(S32, {4});
292 Shape shape = ShapeUtil::MakeShape(F32, {4});
318 Shape shape = ShapeUtil::MakeShape(F32, {4})
[all...] |
| hlo_evaluator_typed_visitor.h | 248 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape()));
258 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape()));
[all...] |
| algebraic_simplifier.h | 44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape);
|
| buffer_value.cc | 31 const Shape& shape = ShapeUtil::GetSubshape(instruction->shape(), index);
|
| call_inliner_test.cc | 60 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
81 const Shape pred = ShapeUtil::MakeShape(PRED, {});
122 const Shape pred = ShapeUtil::MakeShape(PRED, {});
146 const Shape f32 = ShapeUtil::MakeShape(F32, {});
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ir_array.cc | 79 << "Shape " << ShapeUtil::HumanStringWithLayout(shape)
97 << "Shape " << ShapeUtil::HumanStringWithLayout(shape)
103 TF_CHECK_OK(ShapeUtil::ValidateShape(shape));
114 if (!shape_.IsArray() || ShapeUtil::IsScalar(shape_)) {
123 auto b = ShapeUtil::MakeShape(a.element_type(), dims_);
126 ShapeUtil::ElementsIn(a) == ShapeUtil::ElementsIn(b) &&
127 ShapeUtil::ReshapeIsBitcast(a, b);
174 ShapeUtil::ReshapeIsBitcast(input_shape, output_shape)) {
210 ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| call_test.cc | 72 Shape r0f32_ = ShapeUtil::MakeShape(F32, {});
73 Shape r1s0f32_ = ShapeUtil::MakeShape(F32, {0});
74 Shape r1s2f32_ = ShapeUtil::MakeShape(F32, {2});
|
| iota_test.cc | 74 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim);
101 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim);
126 Iota(&builder, ShapeUtil::MakeShape(element_type, dimensions), iota_dim);
|
| replay_test.cc | 74 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(S32, {}), "x");
75 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(S32, {}), "y");
116 Parameter(&plus_two_builder, 0, ShapeUtil::MakeShape(S32, {}), "input");
|
| round_trip_packed_literal_test.cc | 63 reader.Read(ShapeUtil::MakeShape(F32, {2})).ConsumeValueOrDie();
90 Literal actual = reader.Read(ShapeUtil::MakeShape(F32, {2, 2}), &layout)
123 Literal actual = reader.Read(ShapeUtil::MakeShape(F32, {2, 2}), &layout)
|
| dot_operation_test.cc | 192 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 4}), "arg0");
194 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({4, 1}), "arg1");
237 Dot(Parameter(&builder, 0, ShapeUtil::MakeShape(prim_type, {2, 2}), "lhs"),
238 Parameter(&builder, 1, ShapeUtil::MakeShape(prim_type, {2, 2}), "rhs"));
321 ShapeUtil::MakeShapeWithLayout(
326 ShapeUtil::MakeShapeWithLayout(
335 ShapeUtil::MakeShapeWithLayout(
486 Dot(Parameter(&builder, 0, ShapeUtil::MakeShape(prim_type, {2, 3}), "lhs"),
487 Parameter(&builder, 1, ShapeUtil::MakeShape(prim_type, {3, 2}), "rhs"));
517 Dot(Parameter(&builder, 0, ShapeUtil::MakeShape(prim_type, {1, 4}), "lhs")
[all...] |
| literal_test_util.cc | 48 LOG(INFO) << "expected: " << ShapeUtil::HumanString(expected.shape()) << " "
50 LOG(INFO) << "actual: " << ShapeUtil::HumanString(actual.shape()) << " "
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| literal_util.cc | 73 buf_ptrs, xla::ShapeUtil::MakeTupleShape(tensor_shapes));
81 xla::ShapeUtil::ElementsIn(literal.shape()) ==
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 38 return ShapeUtil::ByteSizeOf(shape);
64 XlaBuilder b(absl::StrCat("make_fake_", ShapeUtil::HumanString(shape)));
|
| svd.cc | 119 const int64 n = ShapeUtil::GetDimension(a_shape, -1);
126 batch_dims[k] = ShapeUtil::GetDimension(a_shape, k);
130 auto idx = Iota(builder, ShapeUtil::MakeShape(S32, x_shape.dimensions()),
185 const int64 m = ShapeUtil::GetDimension(a_shape, -2);
192 batch_dims[k] = ShapeUtil::GetDimension(a_shape, k);
196 auto idx = Iota(builder, ShapeUtil::MakeShape(S32, x_shape.dimensions()),
266 batch_dims[i] = ShapeUtil::GetDimension(a_shape, i);
268 const int64 m = ShapeUtil::GetDimension(a_shape, -2);
269 const int64 n = ShapeUtil::GetDimension(a_shape, -1);
471 batch_dims[i] = ShapeUtil::GetDimension(d_shape, i)
[all...] |
| sorting.cc | 31 ShapeUtil::MakeShape(S32, AsInt64Slice(input_shape.dimensions()));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| index_util.cc | 37 << "\n\tshape: " << ShapeUtil::HumanString(shape);
95 DCHECK_LT(linear_index, ShapeUtil::ElementsIn(shape));
|
| literal.h | 219 return ShapeUtil::ElementsIn(shape());
221 return ShapeUtil::ElementsIn(ShapeUtil::GetSubshape(shape(), index));
364 int64 size_bytes() const { return ShapeUtil::ByteSizeOf(subshape()); }
372 : ShapeUtil::ElementsIn(subshape());
712 // a ShapeUtil::ForEachIndex() scan across two literals.
751 Literal() : Literal(ShapeUtil::MakeNil()) {}
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| dumped_computation_to_operation_list.cc | 48 absl::StrAppend(out, ShapeUtil::HumanString(operand->shape()));
53 ShapeUtil::HumanString(hlo->shape()), path_);
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| xrt_state.cc | 103 << xla::ShapeUtil::HumanStringWithLayout(shape) << " device_shape="
104 << xla::ShapeUtil::HumanStringWithLayout(on_device_shape);
113 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first);
243 if (!xla::ShapeUtil::Equal(literal.shape(), on_host_shape())) {
246 xla::ShapeUtil::HumanStringWithLayout(literal.shape()),
247 " device=", xla::ShapeUtil::HumanStringWithLayout(on_host_shape()));
300 xla::ShapeUtil::TryGetSubshape(parent->on_host_shape(), subshape));
303 xla::ShapeUtil::TryGetSubshape(parent->on_device_shape(), subshape));
364 *xla::ShapeUtil::GetMutableSubshape(host_shape, index) =
366 *xla::ShapeUtil::GetMutableSubshape(device_shape, index)
[all...] |
