Searched refs:CopyElementFrom (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1792 result.CopyElementFrom(operand, input_index, output_index)); in HandleGather()
2316 arg_values[i].CopyElementFrom(*input_args[i], input_index, {})); in PerformReductionStep()
2318 accumulators[i].CopyElementFrom(results[i], output_index, {})); in PerformReductionStep()
2342 results[i].CopyElementFrom(computed_results[i], {}, output_index)); in PerformReductionStep()
2346 results[0].CopyElementFrom(computed_result, {}, output_index)); in PerformReductionStep()
2375 results[i].CopyElementFrom(*init_values[i], {}, output_index)); in GenerateReduceOutputElement()
|
| D | hlo_evaluator_typed_visitor.h | 2008 TF_CHECK_OK(curr_val_literal_vec.back().CopyElementFrom(
2043 TF_RETURN_IF_ERROR(results[i].CopyElementFrom(
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 639 Status CopyElementFrom(const LiteralSlice& src_literal,
|
| D | literal.cc | 328 Status MutableLiteralBase::CopyElementFrom(const LiteralSlice& src_literal, in CopyElementFrom() function in xla::MutableLiteralBase
|