Searched refs:GetUniqueSlice (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 271 return *buffers.GetUniqueSlice(hlo, index).ConsumeValueOrDie().allocation(); in GetAllocation()
1716 assignment->GetUniqueSlice(tuple_param0, /*index=*/{0}) in TEST_F()
1718 assignment->GetUniqueSlice(tuple_param1, /*index=*/{0}) in TEST_F()
2107 EXPECT_EQ(assignment->GetUniqueSlice(input0, {}).ConsumeValueOrDie(), in TEST_F()
2108 assignment->GetUniqueSlice(while0, {0}).ConsumeValueOrDie()); in TEST_F()
2110 EXPECT_EQ(assignment->GetUniqueSlice(weights0, {}).ConsumeValueOrDie(), in TEST_F()
2111 assignment->GetUniqueSlice(while0, {1}).ConsumeValueOrDie()); in TEST_F()
2113 EXPECT_EQ(assignment->GetUniqueSlice(while0, {2}).ConsumeValueOrDie(), in TEST_F()
2114 assignment->GetUniqueSlice(while1, {0}).ConsumeValueOrDie()); in TEST_F()
2116 EXPECT_EQ(assignment->GetUniqueSlice(weights1, {}).ConsumeValueOrDie(), in TEST_F()
[all …]
|
| D | buffer_assignment.cc | 472 StatusOr<BufferAllocation::Slice> BufferAssignment::GetUniqueSlice( in GetUniqueSlice() function in xla::BufferAssignment
506 return GetUniqueSlice(instruction, /*index=*/{}); in GetUniqueTopLevelSlice()
512 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() == in SharesSliceAtIndex()
513 GetUniqueSlice(hlo_b, shape_index_b).ConsumeValueOrDie(); in SharesSliceAtIndex()
|
| D | buffer_assignment.h | 417 StatusOr<BufferAllocation::Slice> GetUniqueSlice(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 402 assignment_.GetUniqueSlice(infeed, {0})); in HandleInfeed()
405 assignment_.GetUniqueSlice(infeed, {1})); in HandleInfeed()
420 assignment_.GetUniqueSlice(infeed, {0, i})); in HandleInfeed()
1086 assignment_.GetUniqueSlice(crs, {i})); in HandleAllReduceSingleReplica()
1148 assignment_.GetUniqueSlice(crs, {i})); in HandleAllReduceMultipleReplica()
1158 assignment_.GetUniqueSlice(crs->operand(0), {})); in HandleAllReduceMultipleReplica()
1160 assignment_.GetUniqueSlice(crs, {})); in HandleAllReduceMultipleReplica()
1227 assignment_.GetUniqueSlice(instruction, {i})); in HandleAllToAll()
1274 assignment_.GetUniqueSlice(crs->operand(0), {})); in HandleCollectivePermute()
1278 assignment_.GetUniqueSlice(crs, {})); in HandleCollectivePermute()
[all …]
|
| D | cpu_executable.cc | 255 this->assignment_->GetUniqueSlice(src, value_source->index())); in CreateResultShapedBuffer()
|
| D | ir_emitter.h | 181 return assignment_.GetUniqueSlice(&hlo, index).ConsumeValueOrDie();
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis.cc | 44 auto unique_slice = assignment_.GetUniqueSlice(&hlo, index); in AddAliasingInformationToIrArray()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 344 return ir_emitter_context_->buffer_assignment().GetUniqueSlice(&hlo, index); in MaybeGetAllocationSlice()
|
| D | gpu_executable.cc | 612 assignment.GetUniqueSlice(src_hlo, sources.values()[0]->index())); in GetOutputInfo()
|
| D | ir_emitter.cc | 156 ir_emitter_context_->buffer_assignment().GetUniqueSlice(instr, {}); in EmitConstants()
|
| D | ir_emitter_unnested.cc | 1751 buffer_assignment.GetUniqueSlice(hlo, {})); in GetMlirEmitterInput()
3253 auto slice = buffer_assn.GetUniqueSlice(instr, index); in GetHloBufferSlices()
3265 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
3279 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
3298 auto slice = buffer_assn.GetUniqueSlice(instr, new_index); in GetHloBufferSlices()
3762 buffer_assignment.GetUniqueSlice(a, index).ConsumeValueOrDie(); in CheckHloBuffersShareAllocation()
3764 buffer_assignment.GetUniqueSlice(b, index).ConsumeValueOrDie(); in CheckHloBuffersShareAllocation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1353 assignment_.GetUniqueSlice(instr, shape_index)); in GetOrCreateArrayView()
|