Home
last modified time | relevance | path

Searched refs:data_rank (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_deterministic_test.py42 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument
44 data_dims = data_rank * (data_dim,)
75 data_rank=[1, 2, 3],
79 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
86 hash(data_layout) % 256 + hash(data_rank) % 256 +
92 input_shape = self._makeShapeTuple(batch_size, channel_count, data_rank,
/external/tensorflow/tensorflow/lite/kernels/
Dsegment_sum.cc55 const int data_rank = NumDimensions(data); in ResizeOutputTensor() local
58 for (int i = 1; i < data_rank; ++i) { in ResizeOutputTensor()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h300 std::vector<int64> group_dim_sizes, int64 data_rank, in GroupedSharding()
305 data_rank(data_rank), in GroupedSharding()
310 int64 data_rank; member
Dspmd_partitioner_util.cc1415 grouped.data_rank--; in GroupShardingOnDims()
1440 tiling_dims = std::vector<int64>(grouped_sharding.data_rank, 1); in UngroupSharding()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2020-003.md19 for (int i = 1; i < data_rank; ++i) {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_helpers.inc218 Value cond, int data_rank) {
222 if (cond_tensor.getRank() != 1 || data_rank <= 1) {
229 // length of the shape is equal to data_rank.
230 SmallVector<int64_t, 8> shape(data_rank, 1);
233 RankedTensorType::get({data_rank}, builder->getIntegerType(64));
Dtf_ops_n_z.cc755 int data_rank; in Verify() local
762 data_rank = then_tensor.getRank(); in Verify()
769 data_rank = then_tensor.getRank(); in Verify()
773 data_rank = else_tensor.getRank(); in Verify()
785 if (cond_rank == 0 || cond_rank == data_rank) return success(); in Verify()
789 if (data_rank == 0) { in Verify()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc757 const int64 data_rank = hlo.operand(2)->shape().rank(); in ScatterEffectiveDataSharding() local
758 std::vector<int64> tile_assignment_dims(data_rank, 1LL); in ScatterEffectiveDataSharding()
762 CHECK_LT(i, data_rank); in ScatterEffectiveDataSharding()
787 std::vector<int64> slice_starts(data_rank, 0LL); in ScatterEffectiveDataSharding()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc823 const int32 data_rank = c->Rank(data); in __anon725c8f640402() local
840 if (data_rank == 0) { in __anon725c8f640402()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4872 int64_t data_rank = data_type.getRank(); in matchAndRewrite() local
4907 GetI64ElementsAttrForSeq(segment_ids_rank, data_rank, &rewriter), in matchAndRewrite()