| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference.cc | 235 int64_t operand_index, HloInstruction* dynamic_size) { in DefaultAction()
247 int64_t operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement()
259 int64_t operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
269 int64_t operand_index, HloInstruction* dynamic_size) { in HandleBroadcast()
280 HloInstruction* dynamic_size = in HandleCustomCall() local
322 int64_t operand_index, HloInstruction* dynamic_size) { in HandleCustomCall()
377 int64_t operand_index, HloInstruction* dynamic_size) { in HandleSort()
393 int64_t operand_index, HloInstruction* dynamic_size) { in HandlePad()
447 int64_t operand_index, HloInstruction* dynamic_size) { in HandleReduce()
496 HloInstruction* dynamic_size) { in HandleDot()
[all …]
|
| H A D | dynamic_padder.cc | 182 HloInstruction* dynamic_size = in ReplaceGetSize() local
290 HloInstruction* dynamic_size, in PadWithScalar()
368 HloInstruction* dynamic_size = output_dynamic_dims[output_dim]; in GenerateBinaryMask() local
835 HloInstruction* dynamic_size = in RewriteReverse() local
875 HloInstruction* dynamic_size = in RewriteReverse() local
1321 HloInstruction* dynamic_size = in RewriteDynamicConcat() local
1345 HloInstruction* dynamic_size = nullptr; in RewriteDynamicSort() local
1722 HloInstruction* dynamic_size = in RewriteDynamicReshape() local
|
| H A D | dynamic_padder_test.cc | 69 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
74 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
588 Literal dynamic_size = LiteralUtil::CreateR0<int32_t>(2); in XLA_TEST_F() local
693 Literal dynamic_size = LiteralUtil::CreateR0<int32_t>(1); in XLA_TEST_F() local
794 Literal dynamic_size = LiteralUtil::CreateR0<int32_t>(2); in XLA_TEST_F() local
|
| H A D | dynamic_dimension_inference_test.cc | 1268 auto dynamic_size = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
|
| H A D | hlo_evaluator.cc | 1054 HloInstruction* dynamic_size = in HandleGetDimensionSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | tensor_array_ops.py | 54 dynamic_size=None, argument
403 dynamic_size=None, argument
666 dynamic_size=None, argument
1022 dynamic_size=None, argument
1121 def dynamic_size(self): member in TensorArray
1357 dynamic_size=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | slice_op.cc | 116 auto dynamic_size = in Compile() local
178 xla::XlaOp dynamic_size = in Compile() local
|
| H A D | sequence_ops.cc | 130 auto dynamic_size = (xla::Abs(limit - start) + xla::Abs(delta) - in Compile() local
136 auto dynamic_size = (xla::Ceil(xla::Abs((limit - start) / delta))); in Compile() local
|
| H A D | unique_op.cc | 223 auto dynamic_size = xla::ReduceAll( in CompileWithAxis() local
|
| H A D | tensor_array_ops.cc | 138 bool dynamic_size; in TensorArrayOp() local
|
| /aosp_15_r20/packages/modules/Bluetooth/system/gd/packet/parser/fields/ |
| D | body_field.cc | 37 std::string dynamic_size = "(" + size_field_->GetName() + " * 8)"; in GetSize() local
|
| D | payload_field.cc | 45 std::string dynamic_size = in GetSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_array_ops.cc | 1086 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1268 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
|
| H A D | tensor_array.h | 140 bool identical_element_shapes, bool dynamic_size, in TensorArray()
|
| /aosp_15_r20/hardware/google/gchips/gralloc4/src/hidl_common/ |
| D | BufferDescriptor.h | 127 size_t dynamic_size = 0; in grallocEncodeBufferDescriptor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | structure.py | 53 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/core/ |
| H A D | subgraph.h | 391 size_t dynamic_size; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal.cc | 937 int64_t dynamic_size = GetDynamicSize(i); in Broadcast() local
1043 int64_t dynamic_size = GetDynamicSize(dnum) - start_indices[dnum]; in SliceInternal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| H A D | tensor_array_ops_test.py | 968 def _testWhileLoopWritePackGradients(self, dynamic_size, dtype): argument
|
| /aosp_15_r20/external/mesa3d/src/freedreno/vulkan/ |
| H A D | tu_descriptor_set.cc | 735 uint64_t bo_size = 0, dynamic_size = 0; in tu_CreateDescriptorPool() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 907 auto dynamic_size = handle_op->getAttrOfType<BoolAttr>("dynamic_size"); in matchAndRewrite() local
|