Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression.cc61 e.kind_ = Kind::kTensorList; in TensorList()
85 case Kind::kTensorList: in HumanString()
99 case Kind::kTensorList: in AsXlaOp()
126 case Kind::kTensorList: in ResolveDynamism()
164 case Kind::kTensorList: in ResolveConstant()
214 case Kind::kTensorList: in GetShape()
Dxla_argument.h53 kTensorList, enumerator
Dxla_expression.h52 kTensorList, enumerator
Dgraph_compiler.cc99 case XlaExpression::Kind::kTensorList: { in PrepareArguments()
100 arg.kind = XlaCompiler::Argument::kTensorList; in PrepareArguments()
Dxla_compiler.cc213 case XlaExpression::Kind::kTensorList: { in BuildComputation()
467 case kTensorList: in HumanString()
865 case XlaCompiler::Argument::kTensorList: { in XLAShapeForArgument()
997 case XlaCompiler::Argument::kTensorList: in BuildArguments()
1168 case XlaCompiler::Argument::kTensorList: { in BuildArguments()
Dxla_compiler_test.cc1614 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()