/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()
|