| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | squeeze_spmd_expander.cc | 34 std::set<int64_t> squeeze_dims; in GetSqueezeDims() local
56 std::set<int64_t> squeeze_dims = GetSqueezeDims(op, /*rank=*/shape.size()); in ComputeLayoutForward() local
86 std::set<int64_t> squeeze_dims = GetSqueezeDims(op, /*rank=*/shape.size()); in ComputeLayoutBackward() local
126 std::set<int64_t> squeeze_dims = in ExpandOp() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_squeeze_op.py | 128 squeeze_dims=None): argument
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | util.py | 72 …squeeze_dims = lambda: confusion_matrix.remove_squeezable_dimensions( # pylint: disable=g-long-la… function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 48 const int* squeeze_dims = op_context.params->squeeze_dims; in Prepare() local
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 196 squeeze_dims = lambda: remove_squeezable_dimensions( # pylint: disable=g-long-lambda function
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 201 auto squeeze_dims = squeeze_op.squeeze_dims(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | shape_ops_test.py | 272 def _compareSqueeze(self, x, squeeze_dims, use_gpu): argument
285 def _compareSqueezeAll(self, x, squeeze_dims=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.h | 196 std::vector<int32> squeeze_dims; in SqueezeOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 305 std::vector<int32> squeeze_dims; in SqueezeOp() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1110 auto rebuild_node_def = [&op](const std::vector<int32>& squeeze_dims) { in TEST()
|
| D | array_ops.cc | 2173 std::vector<int32> squeeze_dims; in __anon091a71b73002() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 392 int squeeze_dims[8]; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4551 def squeeze(input, axis=None, name=None, squeeze_dims=None): argument
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1164 auto squeeze_dims = builder->CreateVector(op.squeeze_dims); in WriteOptions() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1740 std::vector<int> squeeze_dims; in ProcessSqueezeOperator() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 1799 AttrValue squeeze_dims; in UpdateSqueezeDims() local
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 2179 const auto* squeeze_dims = schema_params->squeeze_dims(); in ParseSqueeze() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 755 SmallVector<int32_t> squeeze_dims; in matchAndRewrite() local
|
| D | legalize_common.cc | 1245 SmallVectorImpl<int32_t>& squeeze_dims) { in convertSqueezeOp()
|
| D | legalize_tfl.cc | 1744 SmallVector<int32_t> squeeze_dims; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 966 int32_t squeeze_dims[4]; member
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1866 auto& squeeze_dims = (*new_op->mutable_attr())["squeeze_dims"]; in ConvertSqueezeOperator() local
|
| D | import_tensorflow.cc | 1171 const auto& squeeze_dims = GetListAttr(node, "squeeze_dims"); in ConvertSqueezeOperator() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2466 std::vector<int64_t> squeeze_dims; in ConvertSqueeze() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 6473 SmallVector<int64_t> squeeze_dims; in matchAndRewrite() local
|