Searched refs:CopyElementFrom (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 2538 result.CopyElementFrom(operand, input_index, output_index)); in HandleGather()
3621 arg_values[i].CopyElementFrom(*input_args[i], input_index, {})); in PerformReductionStep()
3623 accumulators[i].CopyElementFrom(results[i], output_index, {})); in PerformReductionStep()
3647 results[i].CopyElementFrom(computed_results[i], {}, output_index)); in PerformReductionStep()
3651 results[0].CopyElementFrom(computed_result, {}, output_index)); in PerformReductionStep()
3680 results[i].CopyElementFrom(*init_values[i], {}, output_index)); in GenerateReduceOutputElement()
|
| D | hlo_evaluator_typed_visitor.h | 1836 TF_CHECK_OK(curr_val_literal_vec.back().CopyElementFrom(
1884 TF_CHECK_OK(results[i].CopyElementFrom(computed_result_vec[i], {},
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 832 Status CopyElementFrom(const LiteralSlice& src_literal,
|
| D | literal.cc | 433 Status MutableLiteralBase::CopyElementFrom( in CopyElementFrom() function in xla::MutableLiteralBase
|