Searched refs:GetInputTensorByName (Results 1 – 2 of 2) sorted by relevance
68 return *XlaExpression::CastExpressionFromTensor(GetInputTensorByName(name)); in InputExpression()84 return GetInputTensorByName(name).shape(); in InputShape()108 const Tensor& tensor = GetInputTensorByName(name); in InputType()601 return ReadVariableInputTensor(GetInputTensorByName(name), type, this, shape, in ReadVariableInput()746 return AssignVariableTensor(GetInputTensorByName(name), type, this, handle, in AssignVariable()795 const Tensor& XlaOpKernelContext::GetInputTensorByName(absl::string_view name) { in GetInputTensorByName() function in tensorflow::XlaOpKernelContext
349 const Tensor& GetInputTensorByName(absl::string_view name);