Searched refs:input_tuple (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_util.cc | 22 /*static*/ HloInstruction* TupleUtil::ExtractPrefix(HloInstruction* input_tuple, in ExtractPrefix() argument
24 CHECK(input_tuple->shape().IsTuple()); in ExtractPrefix()
26 HloComputation* computation = input_tuple->parent(); in ExtractPrefix()
27 const Shape& input_shape = input_tuple->shape(); in ExtractPrefix()
34 input_shape.tuple_shapes(i), input_tuple, i))); in ExtractPrefix()
42 HloInstruction* input_tuple, in AppendSuffix() argument
44 CHECK(input_tuple->shape().IsTuple()); in AppendSuffix()
46 HloComputation* computation = input_tuple->parent(); in AppendSuffix()
47 const Shape& input_shape = input_tuple->shape(); in AppendSuffix()
53 input_shape.tuple_shapes(i), input_tuple, i))); in AppendSuffix()
|
| D | tuple_util.h | 30 static HloInstruction* ExtractPrefix(HloInstruction* input_tuple,
40 HloInstruction* input_tuple,
46 static HloInstruction* Duplicate(HloInstruction* input_tuple) { in Duplicate() argument
47 return ExtractPrefix(input_tuple, input_tuple->shape().tuple_shapes_size()); in Duplicate()
|
| D | triangular_solve_expander.cc | 328 auto input_tuple = in InvertDiagonalBlocks() local
331 auto i = GetTupleElement(input_tuple, 0); in InvertDiagonalBlocks()
332 auto body_out = GetTupleElement(input_tuple, 1); in InvertDiagonalBlocks()
333 auto body_input = GetTupleElement(input_tuple, 2); in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_op.cc | 294 xla::XlaOp input_tuple = xla::Tuple(b, inputs); in Compile() local
296 xla::Conditional(ctx->Input(0), input_tuple, *then_result.computation, in Compile()
297 input_tuple, *else_result.computation); in Compile()
|
| D | case_op.cc | 305 auto input_tuple = xla::Tuple(b, inputs); in Compile() local
309 std::vector<xla::XlaOp>(num_branches, input_tuple)); in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | padding_fifo_queue_test.py | 1574 input_tuple = []
1584 input_tuple.append(np_array)
1586 q.enqueue_many(input_tuple).run()
1591 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| D | fifo_queue_test.py | 597 input_tuple = []
607 input_tuple.append(np_array)
609 q.enqueue_many(input_tuple).run()
614 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 2932 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
2933 auto old_operand = input_tuple->mutable_operand(non_windowed_operand_index); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2944 TF_RETURN_IF_ERROR(input_tuple->ReplaceOperandWithDifferentShape( in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2952 input_tuple, loop, loop->while_condition()->parameter_instruction(0), in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3171 auto input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3177 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3184 input_tuple, loop, loop->while_condition()->parameter_instruction(0), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2460 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local
2463 Py_BuildValue("OOOOOO", op_name, attrs, input_tuple.get(), results,
|