Searched refs:tensor_base (Results 1 – 5 of 5) sorted by relevance
387 if (!op.tensor_base().defined()) { in compute_types()395 if (common_device == kCPU && !op.tensor_base().is_cpu()) { in compute_types()396 common_device = op.tensor_base().device(); in compute_types()433 if (!op.tensor_base().defined()) { in compute_types()486 if (!op.tensor_base().defined()) { in compute_types()494 config.allow_cpu_scalars_ && !op.is_output && op.tensor_base().dim() == 0 && in compute_types()495 op.tensor_base().is_cpu()) { in compute_types()520 TORCH_INTERNAL_ASSERT(op.tensor_base().defined()); in compute_types()532 op.tensor_base().options().dtype(common_dtype_), in compute_types()535 namedinference::propagate_names(op.tensor_base(), names_); in compute_types()[all …]
200 const TensorBase& tensor_base() const { in tensor_base() function335 const TensorBase& tensor_base(int64_t arg) const { in tensor_base() function336 return operands_[arg].tensor_base(); in tensor_base()344 return tensor_base(arg);354 return tensor_base(num_outputs_ + arg);386 return c10::fetch_and_cast<T>(op.tensor_base().scalar_type(), op.data); in scalar_value()
73 dest_t* output_data = iter.tensor_base(0).data_ptr<dest_t>(); in reduced_float_copy_kernel()74 scalar_t* input_data = const_cast<scalar_t*>(iter.tensor_base(1).const_data_ptr<scalar_t>()); in reduced_float_copy_kernel()141 dest_t* output_data = iter.tensor_base(0).data_ptr<dest_t>(); in reduced_float_copy_kernel()142 source_t* input_data = const_cast<source_t*>(iter.tensor_base(1).const_data_ptr<source_t>()); in reduced_float_copy_kernel()289 isComplexType(dtype) && (iter.tensor_base(0).is_conj() != iter.tensor_base(1).is_conj())); in copy_kernel()290 const bool requires_neg = (iter.tensor_base(0).is_neg() != iter.tensor_base(1).is_neg()); in copy_kernel()319 auto self = iter.tensor_base(0); in copy_kernel()
55 auto& output_tensor_base = output_operand.tensor_base(); in abs_kernel()56 auto& input_tensor_base = input_operand.tensor_base(); in abs_kernel()110 abs_function((float*)output_operand.tensor_base().mutable_data_ptr(), in abs_kernel()115 !output_operand.original_tensor_base().is_same(output_operand.tensor_base())) { in abs_kernel()
261 const float* tensor_base = GetTensorData<float>(tensor); in ReInterpretTensor() local262 return reinterpret_cast<T>(tensor_base); in ReInterpretTensor()267 float* tensor_base = GetTensorData<float>(tensor); in ReInterpretTensor() local268 return reinterpret_cast<T>(tensor_base); in ReInterpretTensor()