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
|