Home
last modified time | relevance | path

Searched full:type_as (Results 1 – 25 of 79) sorted by relevance

1234

/external/pytorch/torch/testing/_internal/
Dcommon_nn.py80 # `wrap_functional(lambda i: F.poisson_nll_loss(i, t.type_as(i), reduction='none'))`,
87 # `wrap_functional(lambda i: F.poisson_nll_loss(i, t.type_as(i), reduction='none'))`,
178 lambda i: F.poisson_nll_loss(i, t.type_as(i), reduction='none')),
193 lambda i: F.binary_cross_entropy(i, t.type_as(i), reduction='none')),
209 lambda i: F.binary_cross_entropy(i, t.type_as(i), reduction='none')),
225 lambda i: F.binary_cross_entropy(i, t.type_as(i),
226 weight=weights.type_as(i), reduction='none')),
245 lambda i: F.binary_cross_entropy(i, t.type_as(i),
246 weight=weights.type_as(i), reduction='none')),
264 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduce=False)),
[all …]
/external/deqp-deps/SPIRV-Tools/test/val/
Dval_ray_tracing_test.cpp380 %type_as = OpTypeAccelerationStructureKHR in GenerateRayTraceCode()
381 %as_uc_ptr = OpTypePointer UniformConstant %type_as in GenerateRayTraceCode()
419 %as = OpLoad %type_as %top_level_as in TEST_F()
434 %as = OpLoad %type_as %top_level_as in TEST_F()
460 %as = OpLoad %type_as %top_level_as in TEST_F()
472 %as = OpLoad %type_as %top_level_as in TEST_F()
484 %as = OpLoad %type_as %top_level_as in TEST_F()
496 %as = OpLoad %type_as %top_level_as in TEST_F()
508 %as = OpLoad %type_as %top_level_as in TEST_F()
520 %as = OpLoad %type_as %top_level_as in TEST_F()
[all …]
Dval_ray_query_test.cpp66 %type_as = OpTypeAccelerationStructureKHR in GenerateShaderCode()
93 %ptr_as = OpTypePointer UniformConstant %type_as in GenerateShaderCode()
366 %as = OpLoad %type_as %top_level_as in TEST_F()
389 %as_1 = OpLoad %type_as %top_level_as in TEST_F()
397 %as_2 = OpLoad %type_as %top_level_as in TEST_F()
407 %load = OpLoad %type_as %top_level_as in TEST_F()
431 %load = OpLoad %type_as %top_level_as in TEST_F()
443 %load = OpLoad %type_as %top_level_as in TEST_F()
455 %load = OpLoad %type_as %top_level_as in TEST_F()
467 %load = OpLoad %type_as %top_level_as in TEST_F()
[all …]
/external/angle/third_party/spirv-tools/src/test/val/
Dval_ray_tracing_test.cpp380 %type_as = OpTypeAccelerationStructureKHR in GenerateRayTraceCode()
381 %as_uc_ptr = OpTypePointer UniformConstant %type_as in GenerateRayTraceCode()
419 %as = OpLoad %type_as %top_level_as in TEST_F()
434 %as = OpLoad %type_as %top_level_as in TEST_F()
460 %as = OpLoad %type_as %top_level_as in TEST_F()
472 %as = OpLoad %type_as %top_level_as in TEST_F()
484 %as = OpLoad %type_as %top_level_as in TEST_F()
496 %as = OpLoad %type_as %top_level_as in TEST_F()
508 %as = OpLoad %type_as %top_level_as in TEST_F()
520 %as = OpLoad %type_as %top_level_as in TEST_F()
[all …]
Dval_ray_query_test.cpp69 %type_as = OpTypeAccelerationStructureKHR in GenerateShaderCode()
99 %ptr_as = OpTypePointer UniformConstant %type_as in GenerateShaderCode()
373 %as = OpLoad %type_as %top_level_as in TEST_F()
396 %as_1 = OpLoad %type_as %top_level_as in TEST_F()
404 %as_2 = OpLoad %type_as %top_level_as in TEST_F()
414 %load = OpLoad %type_as %top_level_as in TEST_F()
438 %load = OpLoad %type_as %top_level_as in TEST_F()
450 %load = OpLoad %type_as %top_level_as in TEST_F()
462 %load = OpLoad %type_as %top_level_as in TEST_F()
474 %load = OpLoad %type_as %top_level_as in TEST_F()
[all …]
/external/swiftshader/third_party/SPIRV-Tools/test/val/
Dval_ray_tracing_test.cpp380 %type_as = OpTypeAccelerationStructureKHR in GenerateRayTraceCode()
381 %as_uc_ptr = OpTypePointer UniformConstant %type_as in GenerateRayTraceCode()
419 %as = OpLoad %type_as %top_level_as in TEST_F()
434 %as = OpLoad %type_as %top_level_as in TEST_F()
460 %as = OpLoad %type_as %top_level_as in TEST_F()
472 %as = OpLoad %type_as %top_level_as in TEST_F()
484 %as = OpLoad %type_as %top_level_as in TEST_F()
496 %as = OpLoad %type_as %top_level_as in TEST_F()
508 %as = OpLoad %type_as %top_level_as in TEST_F()
520 %as = OpLoad %type_as %top_level_as in TEST_F()
[all …]
Dval_ray_query_test.cpp66 %type_as = OpTypeAccelerationStructureKHR in GenerateShaderCode()
93 %ptr_as = OpTypePointer UniformConstant %type_as in GenerateShaderCode()
366 %as = OpLoad %type_as %top_level_as in TEST_F()
389 %as_1 = OpLoad %type_as %top_level_as in TEST_F()
397 %as_2 = OpLoad %type_as %top_level_as in TEST_F()
407 %load = OpLoad %type_as %top_level_as in TEST_F()
431 %load = OpLoad %type_as %top_level_as in TEST_F()
443 %load = OpLoad %type_as %top_level_as in TEST_F()
455 %load = OpLoad %type_as %top_level_as in TEST_F()
467 %load = OpLoad %type_as %top_level_as in TEST_F()
[all …]
/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
Dtensor_ops.py67 Args: same as ``torch.Tensor.type_as``.
81 Args: same as ``torch.Tensor.type_as``.
90 Handles ``__torch_function__`` dispatch for the ``torch.Tensor.type_as`` op.
92 Args: same as ``torch.Tensor.type_as``.
104 new_local_shards.append(Shard(shard.tensor.type_as(tensor), shard.metadata))
111 torch.Tensor.type_as,
/external/pytorch/torch/csrc/jit/runtime/
Dsymbolic_script.cpp881 … grad_self = (grad_output * (self > other).type_as(grad_output))._grad_sum_to_size(self_size)
882 … grad_other = (grad_output * (other > self).type_as(grad_output))._grad_sum_to_size(other_size)
889 … grad_self = (grad_output * (self < other).type_as(grad_output))._grad_sum_to_size(self.size())
890 … grad_other = (grad_output * (other < self).type_as(grad_output))._grad_sum_to_size(other.size())
906 return grad_output * (result > 0).type_as(result)
920 … return grad_output * torch.where(self > 0, 1.0, negative_slope).type_as(result), None
939 m = (self > 3.).type_as(result)
949 return torch.where(m, lhs, m.type_as(self))
977 … grad_self = (grad_output * condition.type_as(grad_output))._grad_sum_to_size(self_size)
978 …grad_other = (grad_output * (condition.bitwise_not()).type_as(grad_output))._grad_sum_to_size(othe…
[all …]
/external/pytorch/torch/csrc/jit/passes/
Dpeephole.cpp118 node->matches("aten::type_as(Tensor self, Tensor other) -> Tensor") && in optimizeBlock()
120 // x.type_as(y) == x iff x.type() == y.type() in optimizeBlock()
127 " (x.type_as(y) == x) is replaced with ", in optimizeBlock()
349 // insert a type_as conditional on the tensor shape being a scalar, in FuseAddMM()
376 // we can't use type_as if we don't know the target type (mm), the in FuseAddMM()
382 // We insert the type_as if we're sure that the added element is a in FuseAddMM()
389 graph->insertNode(graph->create(aten::type_as, 1)); in FuseAddMM()
/external/executorch/examples/models/llama/
Drope.py92 return xq_out.type_as(xq), xk_out.type_as(xk)
120 t = torch.arange(end, device=freqs.device, dtype=torch.int64).type_as(
/external/pytorch/torch/distributions/
Duniform.py87 lb = self.low.le(value).type_as(self.low)
88 ub = self.high.gt(value).type_as(self.low)
Dnegative_binomial.py48 self.total_count = self.total_count.type_as(self.probs)
54 self.total_count = self.total_count.type_as(self.logits)
Dbinomial.py61 self.total_count = self.total_count.type_as(self.probs)
67 self.total_count = self.total_count.type_as(self.logits)
/external/pytorch/test/jit/
Dtest_custom_operators.py85 r"aten::type_as\(\) is missing value for argument 'other'.",
88 torch.ops.aten.type_as(torch.ones(5, 5))
Dtest_peephole.py65 return x.type_as(y)
68 FileCheck().check("type_as").run(str(tf.graph))
70 FileCheck().check_not("type_as").run(str(tf.graph))
78 return x.type_as(y)
137 return x.type_as(y)
146 FileCheck().check_not("type_as").run(str(trace.graph))
/external/pytorch/test/cpp_api_parity/
Dutils.py71 …# which is strictly equivalent to `F.binary_cross_entropy(i, t.type_as(i), reduction='none')` in P…
284 # lambda i: F.binary_cross_entropy(i, t.type_as(i),
285 # weight=weights.type_as(i), reduction='none')),
/external/executorch/backends/vulkan/
Dcustom_ops_lib.py227 return xq_out.type_as(xq), xk_out.type_as(xk)
/external/executorch/backends/vulkan/_passes/
Dcustom_ops_defs.py227 return xq_out.type_as(xq), xk_out.type_as(xk)
/external/pytorch/torch/csrc/jit/codegen/fuser/cpu/
Dresource_strings.h10 /*with type_as not checking type of its input, a fusion group can have non-fp32
/external/pytorch/torch/distributed/algorithms/model_averaging/
Dutils.py49 p.data = flat_params[offset : offset + p.numel()].view_as(p).type_as(p)
/external/pytorch/benchmarks/gpt_fast/
Dmodel.py265 output = self._norm(x.float()).type_as(x)
292 return x_out2.type_as(x)
Dmixtral_moe_model.py278 output = self._norm(x.float()).type_as(x)
305 return x_out2.type_as(x)
/external/pytorch/test/
Dtest_jit_fuser.py376 mask = (x == 0).type_as(x)
378 mask = (x != 0).type_as(x)
390 mask = (x > 0).type_as(x)
392 mask = (x < 0).type_as(x)
407 mask = (x >= 0).type_as(x)
409 mask = (x <= 0).type_as(x)
/external/pytorch/test/onnx/
Dtest_pytorch_onnx_onnxruntime_cuda.py123 x = x.type_as(y)

1234