| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 56 AsInt64Slice(shape.dimensions()));
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ir_array.cc | 86 linear_ = Linearize(AsInt64Slice(shape.dimensions()), ir_builder);
127 CommonFactors(AsInt64Slice(input_shape.dimensions()),
128 AsInt64Slice(output_shape.dimensions()));
141 AsInt64Slice(output_shape.dimensions()),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| scatter.cc | 48 xla::AsInt64Slice(indices_shape->dimensions());
50 xla::AsInt64Slice(buffer_shape->dimensions());
|
| util.cc | 34 xla::AsInt64Slice(shape.dimensions()));
125 gtl::ArraySlice<int64> major_dims(xla::AsInt64Slice(shape->dimensions()),
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util.cc | 283 return AsInt64Slice(shape.layout().padded_dimensions());
338 return AsInt64Slice(shape.layout().minor_to_major());
344 return AsInt64Slice(layout.minor_to_major());
|
| shape_util.cc | [all...] |
| literal_util.h | [all...] |
| shape_util_test.cc | 648 F32, {3, 4}, AsInt64Slice(input_layout.minor_to_major())),
650 F32, {6, 2}, AsInt64Slice(output_layout.minor_to_major()))),
|
| util.h | 145 static inline tensorflow::gtl::ArraySlice<int64> AsInt64Slice(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| user_computation.cc | 332 AsInt64Slice(gather_request.window_bounds())));
456 AsInt64Slice(map_request.dimensions())));
493 AsInt64Slice(reduce_request.dimensions()), *to_apply_program_shape));
734 operand->output_shape(), AsInt64Slice(reverse_request.dimensions())));
[all...] |
| layout_assignment.cc | 630 AsInt64Slice(instruction->shape().dimensions()));
643 AsInt64Slice(operand_shape.dimensions()));
[all...] |
| layout_assignment_test.cc | 350 AsInt64Slice(log->shape().layout().minor_to_major());
355 AsInt64Slice(reshape->shape().layout().minor_to_major());
[all...] |
| shape_inference.cc | 596 AsInt64Slice(dimension_numbers.lhs_contracting_dimensions());
598 AsInt64Slice(dimension_numbers.rhs_contracting_dimensions());
600 AsInt64Slice(dimension_numbers.lhs_batch_dimensions());
602 AsInt64Slice(dimension_numbers.rhs_batch_dimensions());
[all...] |
| hlo_evaluator.cc | [all...] |
| algebraic_simplifier.cc | [all...] |
| algebraic_simplifier_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_convolution_rewriter.cc | 384 AsInt64Slice(kernel_spatial_dims)));
|
| ir_emitter_unnested.cc | 663 Permute({0, 2, 1}, AsInt64Slice(reduced_a.dimensions())));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_library_test_base.cc | 177 AsInt64Slice(expected.shape().dimensions()), minor_to_major);
|
| reduce_window_test.cc | 621 AsInt64Slice(expected_literal->shape().dimensions()), param.layout);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | [all...] |
