Searched defs:shape_values (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 187 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in CreateTRTEngine() local
371 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 57 llvm::SmallVector<Value, 4> shape_values; in CalculateShapeValue() local
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf.cc | 85 std::vector<int32> shape_values( in InferFilterRank() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1073 def _total_size(shape_values): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to_test.cc | 46 std::initializer_list<ShapeType> shape_values) { in BroadcastToOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 432 SmallVector<Value, 4> shape_values; in GatherShapeInferImpl() local
1629 auto& shape_values = all_shape_values[0]; in reifyReturnTypeShapes() local
1954 SmallVector<Value, 4> shape_values; in reifyReturnTypeShapes() local
2444 SmallVector<Value, 4> shape_values; in reifyReturnTypeShapes() local
2822 SmallVector<Value, 4> shape_values; in reifyReturnTypeShapes() local
3745 SmallVector<Value, 4> shape_values(permutation.size()); in reifyReturnTypeShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4930 SmallVector<Value, 4> shape_values; in matchAndRewrite() local
5824 SmallVector<Value, 4> shape_values; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2836 const auto& shape_values = shape.flat<int32>(); in CreateUpdatedShapeProto() local
|