Searched refs:update_literal (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1301 const Literal& update_literal = parent_->GetEvaluatedLiteralFor(update); in HandleDynamicUpdateSlice() local
1309 DynamicUpdateSlice<int32>(operand_literal, update_literal, in HandleDynamicUpdateSlice()
1315 DynamicUpdateSlice<int64>(operand_literal, update_literal, in HandleDynamicUpdateSlice()
1321 DynamicUpdateSlice<uint32>(operand_literal, update_literal, in HandleDynamicUpdateSlice()
1327 DynamicUpdateSlice<uint64>(operand_literal, update_literal, in HandleDynamicUpdateSlice()
1921 const Literal& operand_literal, const Literal& update_literal, in DynamicUpdateSlice() argument
1935 update_literal.Get<ReturnT>(update_index)); in DynamicUpdateSlice()
1939 std::vector<int64> base(update_literal.shape().dimensions_size(), 0); in DynamicUpdateSlice()
1940 std::vector<int64> step(update_literal.shape().dimensions_size(), 1); in DynamicUpdateSlice()
1941 ShapeUtil::ForEachIndex(update_literal.shape(), base, in DynamicUpdateSlice()
[all …]
|