Searched defs:last_dim (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | applications_test.py | 98 def test_application_notop(self, app, last_dim): argument
112 def test_application_pooling(self, app, last_dim): argument
118 def test_application_variable_input_channels(self, app, last_dim): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 42 int last_dim = input_shape.dims() - 1; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 32 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
80 int last_dim = input_shape.dimensions_size() - 1; in TopKWithPartitions() 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
|
| /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 __anonad10f3360202() local
144 DimensionHandle last_dim = c->Dim(input, 3); in __anonad10f3360302() local
1310 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local
1366 DimensionHandle last_dim = c->Dim(input, -1); in __anonad10f3362002() local
1626 DimensionHandle last_dim = c->Dim(input, 3); in __anonad10f3362a02() local
|
| D | image_ops.cc | 136 DimensionHandle last_dim; in ColorspaceShapeFn() local
|
| D | array_ops.cc | 3302 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon42d741194c02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1083 int64_t last_dim = -1; in matchAndRewrite() local
1084 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/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3769 const int last_dim = in Softmax() local
3911 const int last_dim = in SoftmaxInt8LUT() local
4121 const int last_dim = in LogSoftmax() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3185 int last_dim = std::max(static_cast<int>(input_shape.size()) - 1, 0); in rewriteWithUnknownBegin() local
|