Home
last modified time | relevance | path

Searched refs:kTensorList (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression.cc62 e.kind_ = Kind::kTensorList; in TensorList()
86 case Kind::kTensorList: in HumanString()
100 case Kind::kTensorList: in AsXlaOp()
126 case Kind::kTensorList: in ResolveDynamism()
162 case Kind::kTensorList: in ResolveConstant()
227 case Kind::kTensorList: in GetShape()
Dxla_argument.h53 kTensorList, enumerator
Dxla_expression.h53 kTensorList, enumerator
Dgraph_compiler.cc99 case XlaExpression::Kind::kTensorList: { in PrepareArguments()
100 arg.kind = XlaCompiler::Argument::kTensorList; in PrepareArguments()
Dxla_compiler.cc217 case XlaExpression::Kind::kTensorList: { in BuildComputation()
466 case kTensorList: in HumanString()
887 case XlaCompiler::Argument::kTensorList: { in XLAShapeForArgument()
1024 case XlaCompiler::Argument::kTensorList: in BuildArguments()
1185 case XlaCompiler::Argument::kTensorList: { in BuildArguments()
Dxla_compiler_test.cc1693 args[0].kind = XlaCompiler::Argument::kTensorList; in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc38 case XlaExpression::Kind::kTensorList: { in Compile()
Dtensor_list_utils.cc112 return ctx->InputExpression(index).kind() == XlaExpression::Kind::kTensorList; in IsTensorListInput()