Home
last modified time | relevance | path

Searched refs:GetTensorListBuffer (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc56 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(0), &sum)); in Compile()
62 OP_REQUIRES_OK(ctx, GetTensorListBuffer(ctx->Input(i), &operand)); in Compile()
Dtensor_list_utils.h48 Status GetTensorListBuffer(xla::XlaOp list, xla::XlaOp* buffer);
Dtensor_list_ops.cc408 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()
Dtensor_list_utils.cc150 Status GetTensorListBuffer(xla::XlaOp list, xla::XlaOp* buffer) { in GetTensorListBuffer() function