| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_test.py | 94 def test_application_notop(self, app, last_dim): argument
108 def test_application_pooling(self, app, last_dim): argument
114 def test_application_variable_input_channels(self, app, last_dim): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 32 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
84 int last_dim = input_shape.dimensions_size() - 1; in TopKWithPartitions() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 42 int last_dim = input_shape.dims() - 1; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_bias_add_op.cc | 52 const auto last_dim = input.shape().dims() - 1; in Compute() local
|
| D | nth_element_op.cc | 95 const int last_dim = input_tensor.dim_size(input_tensor.dims() - 1); in operator ()() local
|
| D | quantized_batch_norm_op.cc | 241 const auto last_dim = input.shape().dims() - 1; in Compute() local
|
| /external/tensorflow/tensorflow/c/kernels/ops/ |
| D | bitcast.cc | 43 TF_DimensionHandle* last_dim = TF_NewDimensionHandle(); in ComputeNewShape() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 110 DimensionHandle last_dim = c->Dim(input, 3); in __anon97b5e01b0202() local
144 DimensionHandle last_dim = c->Dim(input, 3); in __anon97b5e01b0302() local
1310 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local
1366 DimensionHandle last_dim = c->Dim(input, -1); in __anon97b5e01b2002() local
1608 DimensionHandle last_dim = c->Dim(input, 3); in __anon97b5e01b2802() local
|
| D | image_ops.cc | 136 DimensionHandle last_dim; in ColorspaceShapeFn() local
|
| D | array_ops.cc | 3298 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon5ce23fde4b02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 945 int64_t last_dim = -1; in matchAndRewrite() local
947 auto is_last_dim_compatible = [](const Value &v, int64_t &last_dim) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 2103 int last_dim = (*dims)[dims_size - 1]; in ProcessMatrixDiagOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 2328 int last_dim = rank - 1; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3778 const int last_dim = in Softmax() local
3920 const int last_dim = in SoftmaxInt8LUT() local
4130 const int last_dim = in LogSoftmax() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3869 int last_dim = std::max(static_cast<int>(input_shape.size()) - 1, 0); in rewriteWithUnknownBegin() local
|