Searched refs:tuple_shapes_size (Results 1 – 16 of 16) sorted by relevance
/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() 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()
|
D | while_loop_simplifier.cc | 329 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 …]
|
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 | dynamic_dimension_inference.cc | 467 int64 operand_count = hlo->shape().tuple_shapes_size(); in HandleWhile()
|
D | dynamic_dimension_inference_test.cc | 531 EXPECT_EQ(while_hlo->shape().tuple_shapes_size(), 3); in TEST_F()
|
D | shape_inference.cc | 2509 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/ |
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 | 340 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()
|
D | shape.h | 112 int tuple_shapes_size() const { return tuple_shapes_.size(); } in tuple_shapes_size() function
|
D | literal_comparison.cc | 834 for (int i = 0; i < expected.tuple_shapes_size(); ++i) { in EqualShapes()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_batchnorm_thunk.cc | 155 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/ |
D | codegen.cc | 215 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/ |
D | cpu_transfer_manager.cc | 206 for (int64 i = 0; i < literal_shape.tuple_shapes_size(); ++i) { in TransferLiteralFromOutfeed()
|
D | ir_emitter.cc | 354 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/ |
D | llvm_util.cc | 227 result_type = llvm::ArrayType::get(result_type, shape.tuple_shapes_size()); in ShapeToIrType()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 334 if (shape->tuple_shapes_size() != 0) { in Build() 335 for (int64 i = 0; i < shape->tuple_shapes_size(); ++i) { in Build()
|