| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_sharding_metadata.cc | 238 ShapeUtil::GetLeafCount(instruction->shape()) ==
|
| call_graph_test.cc | 87 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param0,
92 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {});
|
| hlo_instructions.h | [all...] |
| hlo_alias_analysis.cc | 459 ShapeUtil::ForEachSubshape(
|
| hlo_instruction.h | [all...] |
| hlo_rematerialization.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| sort_util.cc | 322 Shape iteration_shape = ShapeUtil::MakeShape(keys_shape.element_type(),
|
| llvm_util.cc | 253 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape));
500 return ShapeUtil::ByteSizeOf(shape, pointer_size);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| broadcast_simple_test.cc | 61 *r3_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
75 *r2_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major);
|
| dynamic_ops_test.cc | [all...] |
| array_elementwise_ops_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/aot/ |
| codegen.cc | 666 {"{{PROGRAM_SHAPE}}", xla::ShapeUtil::HumanString(xla::ProgramShape(ps))},
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| conv_op_helpers.cc | 108 xla::Shape iota_shape = xla::ShapeUtil::MakeShape(xla::S32, iota_dimensions);
|
| pooling_ops.cc | 579 const xla::Shape scalar = xla::ShapeUtil::MakeShape(xla::F32, {});
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_schedule_test.cc | 39 Shape f32_2x2_ = ShapeUtil::MakeShape(F32, {2, 2});
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_compiler.cc | 509 return ShapeUtil::ByteSizeOf(shape, sizeof(void*));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.h | [all...] |
