| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 51 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()
136 i < new_while_shape.tuple_shapes_size(); i++) { in MakeInstructionsLiveIn()
192 for (int64 i = 1, e = loop_state_shape.tuple_shapes_size(); i < e; i++) { in MakeCountedLoopBodyComputation()
|
| D | while_loop_simplifier.cc | 369 for (int index = 0; index < while_init->shape().tuple_shapes_size(); in TryRemoveRepeatedWhileTupleIndicesHelper()
409 while (index_to_investigate < while_init->shape().tuple_shapes_size()) { in TryRemoveRepeatedWhileTupleIndices()
443 i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveRepeatedWhileTupleIndices()
510 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
527 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
548 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
563 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
804 for (int64 i = 0; i < desired_shape.tuple_shapes_size(); ++i) { in UnflattenTupleInstr()
842 for (int64 i = 0; i < shape.tuple_shapes_size(); ++i) { in GetFlatTupleElems()
896 for (int64 i = 0; i < flat_shape.tuple_shapes_size(); ++i) { in TryFlattenNestedTuples()
[all …]
|
| D | tuple_util.cc | 49 tuple_elements.reserve(input_shape.tuple_shapes_size()); in AppendSuffix()
50 for (int i = 0; i < input_shape.tuple_shapes_size(); i++) { in AppendSuffix()
|
| D | tuple_util.h | 47 return ExtractPrefix(input_tuple, input_tuple->shape().tuple_shapes_size()); in Duplicate()
|
| D | conditional_simplifier.cc | 348 const int old_tuple_shapes_size = conditional_op->shape().tuple_shapes_size(); in RemoveUnusedTupleElements()
558 for (int i = 0; i < conditional->shape().tuple_shapes_size(); ++i) { in MergeDuplicateTupleElements()
|
| D | dynamic_dimension_inference.cc | 58 wide_parameter, narrow_shape.tuple_shapes_size()); in WidenComputation()
1277 hlo->operand(operand_index)->shape().tuple_shapes_size(); in HandleConditional()
1340 int64 original_tuple_count = hlo->shape().tuple_shapes_size(); in HandleConditional()
1378 new_conditional, hlo->shape().tuple_shapes_size()); in HandleConditional()
1439 const int64 original_tuple_count = hlo->shape().tuple_shapes_size(); in HandleWhile()
|
| D | conditional_code_motion.cc | 664 ? ((use_index >= 0) ? conditional->shape().tuple_shapes_size() - 1 in MoveInstructionIn()
665 : conditional->shape().tuple_shapes_size()) in MoveInstructionIn()
689 for (int64 i = 0; i < old_shape.tuple_shapes_size(); ++i) { in MoveInstructionIn()
|
| D | while_loop_all_reduce_code_motion.cc | 465 new_while_instruction->shape().tuple_shapes_size(), nullptr); in CreateNewWhileResult()
|
| D | dynamic_padder.cc | 1504 for (int64 i = 0; i < inst->shape().tuple_shapes_size(); ++i) { in InsertPadToStaticOnInstruction()
1615 for (int64 i = 0; i < shape.tuple_shapes_size(); ++i) { in ConvertToDynamic()
1655 for (int64 i = 0; i < shape.tuple_shapes_size(); ++i) { in ConvertToStatic()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 39 if (sharding.tuple_shardings_size() != shape.tuple_shapes_size()) { in GetShardedShape()
43 sharding.tuple_shardings_size(), shape.tuple_shapes_size(), in GetShardedShape()
47 for (int i = 0; i < shape.tuple_shapes_size(); ++i) { in GetShardedShape()
218 return input_tuple_shape.tuple_shapes_size(); in GetParametersThatMustBeDonated()
|
| D | pjrt_stream_executor_client.h | 476 on_device_shape_.tuple_shapes_size() == 0; in IsEmptyTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 49 tuple_shapes_.reserve(shape_proto.tuple_shapes_size()); in Shape()
68 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size()); in ToProto()
|
| D | shape_util.cc | 398 CHECK(index < tuple_shape->tuple_shapes_size()); in UpdateTupleShape()
497 return shape.tuple_shapes_size(); in TupleElementCount()
764 return pointer_size * shape.tuple_shapes_size(); in ByteSizeOfTupleIndexTable()
797 if (shape.tuple_shapes_size() > 0) { in ValidateShapeWithOptionalLayoutInternal()
884 new_operands.reserve(original.tuple_shapes_size()); in ChangeElementType()
900 if (!subshape->IsTuple() || i >= subshape->tuple_shapes_size() || i < 0) { in IndexIsValid()
924 i >= return_shape->tuple_shapes_size()) { in TryGetSubshape()
|
| D | shape.h | 128 int tuple_shapes_size() const { return tuple_shapes_.size(); } in tuple_shapes_size() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | tpu_driver.cc | 100 if (shape.tuple_shapes_size() > 0) { in ComputeBytesFromShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.h | 76 contents.reserve(shape.tuple_shapes_size()); in ConvertShapeToType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_rewriter.cc | 157 for (int i = 1; i < batch_norm->shape().tuple_shapes_size(); i++) { in HandleBatchNormTraining()
269 for (int i = 1; i < batch_norm->shape().tuple_shapes_size(); i++) { in HandleBatchNormGrad()
|
| D | ir_emitter_nested.cc | 143 for (int i = 0; i < return_shape.tuple_shapes_size(); i++) { in CodegenNestedComputation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 356 list_dynamic_dims.reserve(list_shape.tuple_shapes_size() - 1); in Compile()
357 for (int64 i = 0; i < list_shape.tuple_shapes_size() - 1; ++i) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 93 int tuple_size = tuple_shape.tuple_shapes_size(); in EmitTupleAllocasAtFunctionEntry()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 228 size_t num_results = ps.result().tuple_shapes_size(); in GenResultMethods()
238 ps.result().tuple_shapes_size(), ")"); in GenResultMethods()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 186 for (int64 i = 0; i < original_shape.tuple_shapes_size(); ++i) { in ReshapeWithCorrectRepresentationAndSharding()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 158 const int num_outputs = program_shape.result().tuple_shapes_size(); in GetBaseAuxiliaryParameterCount()
|
| D | tpu_execute.cc | 153 for (int64 i = 0; i < element_shape.tuple_shapes_size(); ++i) { in FixTupleTableAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 206 for (int64 i = 0; i < literal.shape().tuple_shapes_size(); ++i) { in TransferLiteralFromOutfeed()
|