Searched defs:tensor_view (Results 1 – 8 of 8) sorted by relevance
64 …at::Tensor tensor_view = at::empty({}, at::TensorOptions().dtype(tensor.scalar_type()).device(tens… in fill_resize_deterministic_() local
130 absl::string_view tensor_view(mangled_tensor.data(), mangled_tensor.size()); in matchAndRewrite() local
310 absl::string_view tensor_view(mangled_tensor.data(), mangled_tensor.size()); in ConvertTensorProtoAttr() local
467 auto tensor_view = type->AsTensorViewNV(); in GetTypeInstruction() local
454 auto tensor_view = type->AsTensorViewNV(); in GetTypeInstruction() local
2312 const auto tensor_view = _.FindDef(tensor_view_id); in ValidateCooperativeMatrixLoadStoreTensorNV() local
2295 const auto tensor_view = _.FindDef(tensor_view_id); in ValidateCooperativeMatrixLoadStoreTensorNV() local