Searched refs:GetInputTensorByName (Results 1 – 2 of 2) sorted by relevance
58 return *XlaExpression::CastExpressionFromTensor(GetInputTensorByName(name)); in InputExpression()74 return GetInputTensorByName(name).shape(); in InputShape()99 const Tensor& tensor = GetInputTensorByName(name); in InputType()520 return ReadVariableInputTensor(GetInputTensorByName(name), type, this, shape, in ReadVariableInput()662 return AssignVariableTensor(GetInputTensorByName(name), type, this, handle, in AssignVariable()711 const Tensor& XlaOpKernelContext::GetInputTensorByName(absl::string_view name) { in GetInputTensorByName() function in tensorflow::XlaOpKernelContext
299 const Tensor& GetInputTensorByName(absl::string_view name);