Home
last modified time | relevance | path

Searched refs:TensorListGetItemOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc339 class TensorListGetItemOp : public XlaOpKernel { class
341 explicit TensorListGetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListGetItemOp() function in tensorflow::__anond80ffd040111::TensorListGetItemOp
372 TF_DISALLOW_COPY_AND_ASSIGN(TensorListGetItemOp);
375 REGISTER_XLA_OP(Name("TensorListGetItem"), TensorListGetItemOp);
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc711 : public OpConversionPattern<TF::TensorListGetItemOp> {
715 TF::TensorListGetItemOp op, ArrayRef<Value> operands, in matchAndRewrite()
818 if (llvm::isa<TF::TensorListGetItemOp>(op) || in GetTensorListArgumentsFromWhileOp()
991 TF::TensorListGetItemOp, TF::TensorListLengthOp, in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc136 GET_OPERATION_NAME(TF::TensorListGetItemOp), in AddRewrittenCompositeOps()
Dtensor_list_ops_decomposition.cc688 TF::TensorListGetItemOp get_item, in HandleTensorListGetItemOp()
844 } else if (auto get_item = llvm::dyn_cast<TF::TensorListGetItemOp>(&op)) { in DecomposeTensorListOpsInternal()
Dshape_inference.cc271 TensorListLengthOp, TensorListGetItemOp>(use.getOwner())) in CanInferTensorListElementType()