Home
last modified time | relevance | path

Searched refs:constraint_size (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuilder.py2598 constraint_size = None
2608 constraint_size = RelaxedUnspecConstraint(warn_only=False)
2614 constraint_size = StrictMinMaxConstraint(
2619 constraint_size = RelaxedUnspecConstraint(warn_only=False)
2622 constraint_size = RelaxedUnspecConstraint(warn_only=True)
2626 constraint_size = None
2629 constraint_size, name_ = constraint
2633 constraint_sizes.append(constraint_size)
2639 constraint_size is not None
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_op_registry_test.cc72 EXPECT_EQ(kernels.constraint_size(), 1); in TEST()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dkernel_def.pb.h342 int constraint_size() const; in Swap()
830 inline int KernelDef::constraint_size() const { in constraint_size() function
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsymbolic_shapes.py3951 for src, curr_t, constraint_size, constraint_stride in sources_tensors_constraints:
3955 track_symint(property_source, ss, constraint_size[i])
3959 track_symint(property_source, ss, constraint_size[i])
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc1632 for (int i = 0; i < kernel_def.constraint_size(); ++i) { in KernelsRegisteredForOp()