Searched refs:GetTensorListBuffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | aggregate_ops.cc | 39 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(0), &sum)); in Compile()
45 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(i), &operand)); in Compile()
|
| D | tensor_list_ops.cc | 220 OP_REQUIRES_OK(ctx, GetTensorListBuffer(state, &buffer)); in Compile()
253 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(0), &buffer)); in Compile()
327 TF_RETURN_IF_ERROR(GetTensorListBuffer(input_list, output_list_buffer)); in GetInitializedList()
338 TF_RETURN_IF_ERROR(GetTensorListBuffer(output_list, output_list_buffer)); in GetInitializedList()
443 OP_REQUIRES_OK(ctx, GetTensorListBuffer(state, &ta)); in Compile()
|
| D | tensor_list_utils.cc | 38 Status GetTensorListBuffer(const xla::XlaOp& op, xla::XlaOp* buffer) { in GetTensorListBuffer() function
88 TF_RETURN_IF_ERROR(GetTensorListBuffer(uninitialized_list, &input_buffer)); in InitializeTensorList()
|
| D | tensor_list_utils.h | 39 Status GetTensorListBuffer(const xla::XlaOp& op, xla::XlaOp* buffer);
|