Searched refs:a_buffer (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 57 TF_ASSERT_OK_AND_ASSIGN(auto a_buffer, MakeArray(a_shape, client)); in TEST()
61 ASSERT_EQ(a_buffer->device_memory().size(), 1); in TEST()
65 a_buffer->device_memory()[0], b_buffer->device_memory()[0], in TEST()
67 ShapedBuffer shaped_a = a_buffer->AsShapedBuffer( in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cholesky_thunk.cc | 36 BufferAllocation::Slice a_buffer, in CholeskyThunk() argument
43 a_buffer_(a_buffer), in CholeskyThunk()
|
| D | triangular_solve_thunk.cc | 36 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk() argument
47 a_buffer_(a_buffer), in TriangularSolveThunk()
|
| D | cholesky_thunk.h | 45 BufferAllocation::Slice a_buffer,
|
| D | triangular_solve_thunk.h | 43 const BufferAllocation::Slice& a_buffer,
|
| D | ir_emitter_unnested.cc | 1499 TF_ASSIGN_OR_RETURN(auto a_buffer, in EmitCholeskyThunkFromMlir()
1508 if (operand_buffer != a_buffer) { in EmitCholeskyThunkFromMlir()
1512 /*destination_buffer=*/a_buffer, in EmitCholeskyThunkFromMlir()
1519 input.thunk_info, options, a_buffer, workspace_buffer, info_buffer, in EmitCholeskyThunkFromMlir()
|