Searched refs:GetInputTensorByName (Results 1 – 2 of 2) sorted by relevance
76 return *CastExpressionFromTensor(GetInputTensorByName(name)); in InputExpression()92 return GetInputTensorByName(name).shape(); in InputShape()100 return GetInputTensorByName(name).dtype(); in InputType()416 return ReadVariableInputTensor(GetInputTensorByName(name), type, this, shape, in ReadVariableInput()553 return AssignVariableTensor(GetInputTensorByName(name), type, this, handle, in AssignVariable()592 const Tensor& XlaOpKernelContext::GetInputTensorByName(absl::string_view name) { in GetInputTensorByName() function in tensorflow::XlaOpKernelContext
267 const Tensor& GetInputTensorByName(absl::string_view name);