Searched refs:GetTensorListBuffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | aggregate_ops.cc | 56 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(0), &sum)); in Compile()
62 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(i), &operand)); in Compile()
|
| D | tensor_list_utils.h | 48 Status GetTensorListBuffer(xla::XlaOp list, xla::XlaOp* buffer);
|
| D | tensor_list_ops.cc | 408 OP_REQUIRES_OK(ctx, GetTensorListBuffer(list, &buffer)); in Compile()
450 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(0), &buffer)); in Compile()
481 OP_REQUIRES_OK(ctx, GetTensorListBuffer(input, &buffer)); in Compile()
|
| D | tensor_list_utils.cc | 150 Status GetTensorListBuffer(xla::XlaOp list, xla::XlaOp* buffer) { in GetTensorListBuffer() function
|