| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_array_ops.cc | 63 TF_RET_CHECK(resource->max_array_size() >= 0) in MaybeInitializeTensorArray()
64 << resource->name() << " size " << resource->max_array_size(); in MaybeInitializeTensorArray()
80 ta_shape.AddDim(resource->max_array_size()); in MaybeInitializeTensorArray()
116 shape->InsertDim(0, resource->max_array_size()); in GetTensorArrayShape()
518 ta_shape.AddDim(resource->max_array_size()); in Compile()
521 OP_REQUIRES(ctx, lengths.size() == resource->max_array_size(), in Compile()
524 lengths.size(), " vs. ", resource->max_array_size(), ")")); in Compile()
562 size_tensor.scalar<int32>()() = static_cast<int32>(var->max_array_size()); in Compile()
|
| H A D | stack_ops.cc | 71 stack_shape.AddDim(resource->max_array_size()); in MaybeInitializeStack()
|
| /aosp_15_r20/external/OpenCL-CTS/test_conformance/images/kernel_read_write/ |
| H A D | test_cl_ext_image_requirements_info.cpp | 464 uint32_t max_array_size = 0; in cl_image_requirements_max_val_ext_positive() local
467 CL_IMAGE_REQUIREMENTS_MAX_ARRAY_SIZE_EXT, sizeof(max_array_size), in cl_image_requirements_max_val_ext_positive()
468 &max_array_size, nullptr); in cl_image_requirements_max_val_ext_positive()
476 if (!(max_array_size <= array_size && max_array_size > 0)) in cl_image_requirements_max_val_ext_positive()
|
| /aosp_15_r20/external/gemmlowp/test/ |
| H A D | test_allocator.cc | 20 void test_allocator(Allocator* a, int max_array_size) { in test_allocator() argument
21 const std::size_t int32_array_size = Random() % max_array_size; in test_allocator()
23 const std::size_t int8_array_size = Random() % max_array_size; in test_allocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_resource.cc | 54 int64_t max_array_size) { in CreateTensorArray() argument
57 initial_value, max_array_size, in CreateTensorArray()
64 xla::XlaOp initial_value, int64_t max_array_size, in XlaResource() argument
75 max_array_size_(max_array_size), in XlaResource()
|
| H A D | xla_resource.h | 49 int64_t max_array_size);
53 int64_t max_array_size,
151 int64_t max_array_size() const { return max_array_size_; } in max_array_size() function
|
| H A D | xla_argument.cc | 23 if (std::tie(kind, resource_kind, type, name, initialized, max_array_size, in operator ==()
26 other.initialized, other.max_array_size, in operator ==()
|
| H A D | xla_compiler.cc | 455 if (max_array_size >= 0) { in HumanString()
456 absl::StrAppend(&output, " max_array_size=", max_array_size); in HumanString()
913 if (arg.max_array_size < 0) { in XLAShapeForArgument()
919 shape.AddDim(arg.max_array_size); in XLAShapeForArgument()
931 if (arg.max_array_size < 0) { in XLAShapeForArgument()
937 shape.AddDim(arg.max_array_size); in XLAShapeForArgument()
970 arg->max_array_size = resource.max_array_size(); in PopulateArgumentFromResource()
1010 /*max_array_size=*/arg.max_array_size, in BuildArguments()
|
| H A D | xla_argument.h | 102 int64_t max_array_size = -1; member
|
| H A D | xla_compiler_test.cc | 827 args[0].max_array_size = 2; in TEST_F()
885 args[0].max_array_size = 2; in TEST_F()
917 args[0].max_array_size = 2; in TEST_F()
|
| /aosp_15_r20/external/mesa3d/src/compiler/glsl/ |
| H A D | gl_nir_link_uniforms.c | 120 int max_array_size = 0; in update_array_sizes() local
130 max_array_size = MAX2(BITSET_LAST_BIT_SIZED(ainfo->indices, words), in update_array_sizes()
131 max_array_size); in update_array_sizes()
134 if (max_array_size == glsl_array_size(var->type)) in update_array_sizes()
138 if (max_array_size != glsl_array_size(var->type)) { in update_array_sizes()
150 (max_array_size * (num_slots / glsl_array_size(var->type))); in update_array_sizes()
154 max_array_size, 0); in update_array_sizes()
|
| /aosp_15_r20/external/deqp/external/openglcts/modules/glesext/tessellation_shader/ |
| H A D | esextcTessellationShaderTessellation.cpp | 2660 int max_array_size = de::max(m_gl_max_tess_control_input_components_value, in initReferenceValues() local
2665 m_ref_vertex_attributes = (glw::GLfloat *)malloc(sizeof(glw::GLfloat) * (max_array_size)); in initReferenceValues()
|