HomeSort by relevance Sort by last modified time
    Searched refs:tuple_shapes_size (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
tuple_util.cc 49 tuple_elements.reserve(input_shape.tuple_shapes_size());
50 for (int i = 0; i < input_shape.tuple_shapes_size(); i++) {
while_util.cc 51 narrow_shape.tuple_shapes_size());
75 wide_parameter, narrow_shape.tuple_shapes_size());
81 for (int i = narrow_shape.tuple_shapes_size();
82 i < wide_shape.tuple_shapes_size(); i++) {
102 int64 elements_in_old_while_shape = while_instr->shape().tuple_shapes_size();
129 new_while, while_instr->shape().tuple_shapes_size())));
135 i < new_while_shape.tuple_shapes_size(); i++) {
190 for (int64 i = 1, e = loop_state_shape.tuple_shapes_size(); i < e; i++) {
while_loop_simplifier.cc 329 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) {
346 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) {
367 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) {
382 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) {
592 for (int64 i = 0; i < desired_shape.tuple_shapes_size(); ++i) {
630 for (int64 i = 0; i < shape.tuple_shapes_size(); ++i) {
684 for (int64 i = 0; i < flat_shape.tuple_shapes_size(); ++i) {
    [all...]
dynamic_dimension_inference.cc 467 int64 operand_count = hlo->shape().tuple_shapes_size();
dynamic_dimension_inference_test.cc 531 EXPECT_EQ(while_hlo->shape().tuple_shapes_size(), 3);
shape_inference.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape.cc 49 tuple_shapes_.reserve(shape_proto.tuple_shapes_size());
68 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size());
shape.h 112 int tuple_shapes_size() const { return tuple_shapes_.size(); } function in class:xla::Shape
shape_util.cc 340 return shape.tuple_shapes_size();
596 return pointer_size * shape.tuple_shapes_size();
640 if (shape.tuple_shapes_size() > 0) {
773 if (!subshape->IsTuple() || i >= subshape->tuple_shapes_size() || i < 0) {
797 i >= return_shape->tuple_shapes_size()) {
    [all...]
literal_comparison.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc 155 CHECK_EQ(hlo->shape().tuple_shapes_size(), 3);
237 CHECK_EQ(hlo->shape().tuple_shapes_size(), 3);
  /external/tensorflow/tensorflow/compiler/aot/
codegen.cc 215 size_t num_results = ps.result().tuple_shapes_size();
220 ps.result().tuple_shapes_size(), ")");
267 size_t num_results = ps.result().tuple_shapes_size();
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_transfer_manager.cc 206 for (int64 i = 0; i < literal_shape.tuple_shapes_size(); ++i) {
ir_emitter.cc 354 for (int64 i = 0; i < data_shape.tuple_shapes_size(); ++i) {
491 for (int64 i = 0; i < operand_shape.tuple_shapes_size(); ++i) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
llvm_util.cc 227 result_type = llvm::ArrayType::get(result_type, shape.tuple_shapes_size());
  /external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.cc 334 if (shape->tuple_shapes_size() != 0) {
335 for (int64 i = 0; i < shape->tuple_shapes_size(); ++i) {
    [all...]

Completed in 119 milliseconds