| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_permute_thunk.cc | 42 int64_t partition_count) { in GetNcclCollectivePermuteConfig()
84 int64_t partition_count) { in IsDegenerate()
107 int64_t replica_count, int64_t partition_count, const Buffer& buffer) in NcclCollectivePermuteThunk()
|
| D | nccl_all_reduce_thunk.cc | 161 bool IsDegenerate(OpT op, int64_t replica_count, int64_t partition_count) { in IsDegenerate()
238 int64_t partition_count) { in IsDegenerate()
275 int64_t partition_count) { in IsDegenerate()
350 int64_t partition_count) { in IsDegenerate()
|
| D | nccl_all_gather_thunk.h | 46 int64_t partition_count) { in IsDegenerate()
|
| D | nccl_all_to_all_thunk.h | 46 int64_t partition_count) { in IsDegenerate()
|
| D | ir_emitter_unnested.cc | 2769 const int64_t partition_count = hlo_module_config_.num_partitions(); in EmitCollectivePermute() local
2809 int64_t partition_count = hlo_module_config_.num_partitions(); in EmitNcclThunk() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 134 int partition_count = device_assignment.computation_count(); in GetParticipatingDevicesGroups() local
243 int partition_count = device_assignment.computation_count(); in GetParticipatingDevices() local
|
| D | collective_decomposer_utils.cc | 53 HloInstruction *partition_count = in CreateStartIndicesForCollectiveDecomposition() local
|
| D | reduce_scatter_decomposer_test.cc | 45 const int64_t partition_count = 2; in RunPass() local
|
| D | ar_crs_combiner.cc | 46 int64_t partition_count) { in ReplaceReplicatedAllReduce()
|
| /external/swiftshader/third_party/astc-encoder/Source/ |
| D | astc_codec_internals.h | 66 int partition_count; member
222 int partition_count; // 1 to 4; Zero marks a constant-color block. member
276 int partition_count in get_partition_table()
|
| D | astc_symbolic_physical.cpp | 127 int partition_count = read_bits(2, 11, pb.data) + 1; in physical_to_symbolic() local
|
| D | astc_decompress_symbolic.cpp | 202 int partition_count = scb->partition_count; in decompress_symbolic_block() local
|
| D | astc_partition_tables.cpp | 227 int partition_count, in generate_one_partition_table()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op_gpu.cu.cc | 208 const Tensor* partitions, const Tensor* partition_count, in AllocateOutputs()
236 Tensor partition_count; in ComputeAsync() local
358 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts()
|
| D | dynamic_partition_op.cc | 56 gtl::InlinedVector<int, 32> partition_count(num_partitions_); in ValidateAndAllocateOutputs() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition_test.cc | 185 const int64_t partition_count = partition_iterator.GetTotalPartitionCount(); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1044 auto partition_count = in PartitionBaseCase() local
1424 auto partition_count = in PartitionBaseCase() local
|
| D | spmd_partitioner.cc | 2472 const int64_t partition_count = in HandleSort() local
|
| /external/libbpf/.github/actions/build-selftests/ |
| D | vmlinux.h | 60786 u8 partition_count; member
|
| /external/bcc/libbpf-tools/powerpc/ |
| D | vmlinux_510.h | 92787 u8 partition_count; member
|
| D | vmlinux.h | 92787 u8 partition_count; member
|
| /external/bcc/libbpf-tools/arm64/ |
| D | vmlinux.h | 64752 u8 partition_count; member
|
| D | vmlinux_510.h | 64752 u8 partition_count; member
|
| /external/bcc/libbpf-tools/x86/ |
| D | vmlinux_505.h | 61783 u8 partition_count; member
|
| D | vmlinux.h | 61783 u8 partition_count; member
|