Searched refs:GetDType (Results 1 – 4 of 4) sorted by relevance
115 tensorflow::DataType GetDType() const;
951 if (output_tensor.GetDType() == tensorflow::DT_VARIANT) { in ForwardpropFromTape()964 " with dtype ", output_tensor.GetDType()); in ForwardpropFromTape()1065 if (IsDtypeTrainable(input_tensors[target_index].GetDType())) { in Accumulate()
87 tensorflow::DataType TapeTensor::GetDType() const { in GetDType() function in tensorflow::gradients::TapeTensor
1172 tensorflow::DataType GetDType() const { return dtype_; } in GetDType() function in PyTapeTensor