| /external/tensorflow/tensorflow/compiler/jit/ |
| shape_inference.cc | 72 const InferredShape& arg_shape = it->second; local
76 if (arg_shape.handle_type != DT_INVALID) {
79 arg_shape.handle_shape, &handle));
84 {handle, arg_shape.handle_type}});
89 context->MakeShapeFromPartialTensorShape(arg_shape.shape, &handle));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_cpu.cc | 112 xla::Shape arg_shape = shape_status.ConsumeValueOrDie(); variable
113 *arg_shape.mutable_layout() =
114 xla::LayoutUtil::MakeDescendingLayout(arg_shape.rank());
115 arg_shapes.push_back(std::move(arg_shape));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator.cc | 245 const auto& arg_shape = arg_literals[i]->shape(); local
246 if (!ShapeUtil::Equal(computation_shape, arg_shape)) {
251 ShapeUtil::HumanString(arg_shape));
[all...] |
| shape_inference.cc | 357 const Shape* arg_shape = nullptr; local
361 if (!arg_shape) {
362 arg_shape = shape;
363 element_type = arg_shape->element_type();
366 if (arg_shape->rank() != shape->rank()) {
370 arg_shape->rank(), ShapeUtil::HumanString(*arg_shape), shape->rank(),
373 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shape, *shape)) {
376 PrimitiveType_Name(arg_shape->element_type()),
379 for (int64 dimension_number = 0; dimension_number < arg_shape->rank()
1080 const Shape* arg_shape = arg_shapes[0]; local
2852 const Shape& arg_shape = *arg_shapes[i]; local
[all...] |
| shape_inference_test.cc | 802 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); local
813 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); local
824 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); local
[all...] |
