Home
last modified time | relevance | path

Searched refs:max_array_size (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtensor_array_ops.cc63 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 Dstack_ops.cc71 stack_shape.AddDim(resource->max_array_size()); in MaybeInitializeStack()
/aosp_15_r20/external/OpenCL-CTS/test_conformance/images/kernel_read_write/
H A Dtest_cl_ext_image_requirements_info.cpp464 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 Dtest_allocator.cc20 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 Dxla_resource.cc54 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 Dxla_resource.h49 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 Dxla_argument.cc23 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 Dxla_compiler.cc455 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 Dxla_argument.h102 int64_t max_array_size = -1; member
H A Dxla_compiler_test.cc827 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 Dgl_nir_link_uniforms.c120 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 DesextcTessellationShaderTessellation.cpp2660 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()