Home
last modified time | relevance | path

Searched defs:shaped (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir/utils/runtime/
Dconstraints.cc72 auto shaped = type.dyn_cast<ShapedType>(); in ResolveArgumentConstraint() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize.cc154 auto shaped = arg.getType().dyn_cast<ShapedType>(); in SetInputNodesQuantizationParams() local
180 if (auto shaped = input_type.dyn_cast<ShapedType>()) { in SetInputNodesQuantizationParams() local
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dprepare_quantize.cc168 auto shaped = arg.getType().dyn_cast<ShapedType>(); in SetInputNodesQuantizationParams() local
194 if (auto shaped = input_type.dyn_cast<ShapedType>()) { in SetInputNodesQuantizationParams() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_types.cc81 if (auto shaped = type.dyn_cast<ShapedType>()) { in ToLegalType() local
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Dspecialization.cc49 static llvm::Error VerifyMemrefOperand(unsigned index, mlir::ShapedType shaped, in VerifyMemrefOperand()
/external/tensorflow/tensorflow/core/framework/
Dtensor.h879 typename TTypes<T, NDIMS>::Tensor Tensor::shaped( in shaped() function
906 typename TTypes<T, NDIMS>::ConstTensor Tensor::shaped( in shaped() function
Dtensor_test.cc402 T shaped = (t.*Func)(sizes); in TestReshape() local
408 T shaped = (static_cast<const Tensor&>(t).*Func)(sizes); in TestReshape() local
413 void TestReshapeImpl(T shaped, std::initializer_list<int64_t> sizes) { in TestReshapeImpl()
/external/tensorflow/tensorflow/compiler/xla/runtime/
Dsymbolic_shape.cc157 const MemrefDesc* shaped = cast<MemrefDesc>(&arguments[i]); in ResolveImpl() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc147 auto shaped = target.dyn_cast<RankedTensorType>(); in ResetAxisAndBroadcast() local
/external/skia/modules/skparagraph/src/
DTextLine.cpp684 … auto shaped = [&](sk_sp<SkTypeface> typeface, sk_sp<SkFontMgr> fallback) -> std::unique_ptr<Run> { in shapeEllipsis() local