Home
last modified time | relevance | path

Searched refs:TensorListReserveOp (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc134 class TensorListReserveOp : public XlaOpKernel { class
136 explicit TensorListReserveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListReserveOp() function in tensorflow::__anond80ffd040111::TensorListReserveOp
201 TF_DISALLOW_COPY_AND_ASSIGN(TensorListReserveOp);
207 TensorListReserveOp);
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc482 : public ConvertTensorListInitOp<TF::TensorListReserveOp> {
486 Value GetNumElements(TF::TensorListReserveOp op, ArrayRef<Value> operands, in GetNumElements()
662 auto extended_part = rewriter->create<TF::TensorListReserveOp>( in CreateCondTrueBranch()
821 llvm::isa<TF::TensorListReserveOp>(op) || in GetTensorListArgumentsFromWhileOp()
992 TF::TensorListPushBackOp, TF::TensorListReserveOp, in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc132 GET_OPERATION_NAME(TF::TensorListReserveOp), in AddRewrittenCompositeOps()
Dtensor_list_ops_decomposition.cc590 TF::TensorListReserveOp list, in HandleTensorListReserveOp()
828 } else if (auto list = llvm::dyn_cast<TF::TensorListReserveOp>(&op)) { in DecomposeTensorListOpsInternal()
Dshape_inference.cc139 return isa<TensorListReserveOp>(op) || isa<EmptyTensorListOp>(op) || in IsTensorListInitOp()
147 if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op)) in GetElementShapeOperand()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2177 static LogicalResult Verify(TensorListReserveOp op) { in Verify()