Home
last modified time | relevance | path

Searched refs:tuple_shapes_size (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_util.cc51 narrow_shape.tuple_shapes_size()); in WidenWhileCondition()
75 wide_parameter, narrow_shape.tuple_shapes_size()); in WidenWhileBody()
81 for (int i = narrow_shape.tuple_shapes_size(); in WidenWhileBody()
82 i < wide_shape.tuple_shapes_size(); i++) { in WidenWhileBody()
102 int64 elements_in_old_while_shape = while_instr->shape().tuple_shapes_size(); in MakeInstructionsLiveIn()
129 new_while, while_instr->shape().tuple_shapes_size()))); in MakeInstructionsLiveIn()
135 i < new_while_shape.tuple_shapes_size(); i++) { in MakeInstructionsLiveIn()
190 for (int64 i = 1, e = loop_state_shape.tuple_shapes_size(); i < e; i++) { in MakeCountedLoopBodyComputation()
Dwhile_loop_simplifier.cc329 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
346 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
367 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
382 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
592 for (int64 i = 0; i < desired_shape.tuple_shapes_size(); ++i) { in UnflattenTupleInstr()
630 for (int64 i = 0; i < shape.tuple_shapes_size(); ++i) { in GetFlatTupleElems()
684 for (int64 i = 0; i < flat_shape.tuple_shapes_size(); ++i) { in TryFlattenNestedTuples()
857 trip_counter = new_while_shape.tuple_shapes_size(); in TryMergeInductionVariables()
869 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryMergeInductionVariables()
895 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryMergeInductionVariables()
[all …]
Dtuple_util.cc49 tuple_elements.reserve(input_shape.tuple_shapes_size()); in AppendSuffix()
50 for (int i = 0; i < input_shape.tuple_shapes_size(); i++) { in AppendSuffix()
Ddynamic_dimension_inference.cc467 int64 operand_count = hlo->shape().tuple_shapes_size(); in HandleWhile()
Ddynamic_dimension_inference_test.cc531 EXPECT_EQ(while_hlo->shape().tuple_shapes_size(), 3); in TEST_F()
Dshape_inference.cc2509 if (index < 0 || index >= arg.tuple_shapes_size()) { in InferGetTupleElementShape()
2513 index, arg.tuple_shapes_size(), ShapeUtil::HumanString(arg)); in InferGetTupleElementShape()
/external/tensorflow/tensorflow/compiler/xla/
Dshape.cc49 tuple_shapes_.reserve(shape_proto.tuple_shapes_size()); in Shape()
68 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size()); in ToProto()
Dshape_util.cc340 return shape.tuple_shapes_size(); in TupleElementCount()
596 return pointer_size * shape.tuple_shapes_size(); in ByteSizeOfTupleIndexTable()
640 if (shape.tuple_shapes_size() > 0) { in ValidateShapeWithOptionalLayoutInternal()
773 if (!subshape->IsTuple() || i >= subshape->tuple_shapes_size() || i < 0) { in IndexIsValid()
797 i >= return_shape->tuple_shapes_size()) { in TryGetSubshape()
Dshape.h112 int tuple_shapes_size() const { return tuple_shapes_.size(); } in tuple_shapes_size() function
Dliteral_comparison.cc834 for (int i = 0; i < expected.tuple_shapes_size(); ++i) { in EqualShapes()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc155 CHECK_EQ(hlo->shape().tuple_shapes_size(), 3); in CudnnBatchNormForwardTrainingThunk()
237 CHECK_EQ(hlo->shape().tuple_shapes_size(), 3); in CudnnBatchNormBackwardThunk()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen.cc215 size_t num_results = ps.result().tuple_shapes_size(); in GenResultMethods()
220 ps.result().tuple_shapes_size(), ")"); in GenResultMethods()
267 size_t num_results = ps.result().tuple_shapes_size(); in GenVariableMethods()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc206 for (int64 i = 0; i < literal_shape.tuple_shapes_size(); ++i) { in TransferLiteralFromOutfeed()
Dir_emitter.cc354 for (int64 i = 0; i < data_shape.tuple_shapes_size(); ++i) { in HandleInfeed()
491 for (int64 i = 0; i < operand_shape.tuple_shapes_size(); ++i) { in HandleOutfeed()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc227 result_type = llvm::ArrayType::get(result_type, shape.tuple_shapes_size()); in ShapeToIrType()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc334 if (shape->tuple_shapes_size() != 0) { in Build()
335 for (int64 i = 0; i < shape->tuple_shapes_size(); ++i) { in Build()