Home
last modified time | relevance | path

Searched defs:input0_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dset_ops.cc54 ShapeHandle input0_shape = c->input(0); in __anona97893f50102() local
116 ShapeHandle input0_shape = c->input(0); in __anona97893f50202() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_binary.cc99 Shape input0_shape = input0_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClWinogradConv2d.cpp135 …const TensorShape input0_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*s… in validate_arguments() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dcommon.h962 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast()
995 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp1175 auto input0_shape = ConstantValueMap::GetShape(n->input(0)->debugName()); in ProcessTimeSeriesNode() local