Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc52 static const XlaExpression* CastExpressionFromTensor(const Tensor& tensor) { in CastExpressionFromTensor() function
71 return *CastExpressionFromTensor(context_->input(index)); in InputExpression()
76 return *CastExpressionFromTensor(GetInputTensorByName(name)); in InputExpression()
351 handles->push_back(CastExpressionFromTensor(input)->AsXlaOp(builder())); in InputList()
373 const XlaExpression* expression = CastExpressionFromTensor(tensor); in ReadVariableInputTensor()
423 const XlaExpression* expression = CastExpressionFromTensor(tensor); in GetVariableTypeAndShape()
502 CastExpressionFromTensor(context_->input(index)); in GetResourceInput()
513 const XlaExpression* expression = CastExpressionFromTensor(tensor); in AssignVariableTensor()