| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 297 Shape param_shape = ShapeUtil::MakeShape(F32, {1, 4}); in TEST_F() local
318 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
351 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
370 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
392 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
410 Shape param_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
429 Shape param_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
447 Shape param_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
466 Shape param_shape = ShapeUtil::MakeShape(F32, {3, 4}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| H A D | cpu_eigen_dot_operation_test.cc | 68 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
83 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P() local
|
| H A D | cpu_vectorization_test.cc | 92 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
|
| H A D | cpu_intrinsic_test.cc | 96 auto param_shape = ShapeUtil::MakeShape(F32, {1024}); in TEST_P() local
|
| H A D | cpu_noalias_test.cc | 44 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | gpu_ftz_test.cc | 37 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateBinaryOpModule() local
54 Shape param_shape = ShapeUtil::MakeShapeWithLayout( in CreateUnaryOpModule() local
|
| H A D | gpu_noalias_test.cc | 36 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
| H A D | gpu_ldg_test.cc | 128 auto param_shape = ShapeUtil::MakeShape(F32, {32, 32}); in TEST_F() local
|
| H A D | gpu_index_test.cc | 45 auto param_shape = ShapeUtil::MakeShape(F32, {5, 7, 2}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | test_utils.cc | 611 const HloInstruction& param, const Shape& param_shape, in CreateLiteralForConstrainedUses()
714 const Shape& param_shape, in MakeConstrainedArgument()
746 const Shape& param_shape = (module_config.has_entry_computation_layout() && in MakeFakeArguments() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 683 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
854 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
895 Shape param_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
|
| H A D | hlo_input_output_alias_config.cc | 193 const Shape& param_shape = in Verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_batch_normalization.cc | 106 const auto& param_shape = mean_shape; in Run() local
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | multinomial.py | 98 def param_shape(self): member in Multinomial
|
| H A D | negative_binomial.py | 98 def param_shape(self): member in NegativeBinomial
|
| H A D | categorical.py | 105 def param_shape(self): member in Categorical
|
| H A D | binomial.py | 115 def param_shape(self): member in Binomial
|
| H A D | bernoulli.py | 101 def param_shape(self): member in Bernoulli
|
| H A D | one_hot_categorical.py | 91 def param_shape(self): member in OneHotCategorical
|
| H A D | relaxed_categorical.py | 65 def param_shape(self): member in ExpRelaxedCategorical
|
| H A D | relaxed_bernoulli.py | 89 def param_shape(self): member in LogitRelaxedBernoulli
|
| H A D | continuous_bernoulli.py | 162 def param_shape(self): member in ContinuousBernoulli
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | bincount_op.cc | 142 xla::Shape param_shape = xla::ShapeUtil::MakeShape(dtype, {}); in Compile() local
|
| H A D | where_op.cc | 285 xla::Shape param_shape = xla::ShapeUtil::MakeShape(S32, {}); in CompileWhereWithPrefixSum() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_nested.cc | 72 const Shape& param_shape = param->shape(); in CodegenNestedComputation() local
|