Home
last modified time | relevance | path

Searched refs:input_depth_value (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc673 int64_t input_depth_value = c->Value(input_depth_dim), in Conv2DShapeImpl() local
677 if (input_depth_value % filter_input_depth_value != 0) in Conv2DShapeImpl()
679 "Depth of input (", input_depth_value, in Conv2DShapeImpl()
682 if (input_depth_value != filter_input_depth_value) { in Conv2DShapeImpl()
683 int64_t num_groups = input_depth_value / filter_input_depth_value; in Conv2DShapeImpl()
836 int64_t input_depth_value = c->Value(input_depth_dim), in Conv3DShape() local
840 if (input_depth_value % filter_input_depth_value != 0) in Conv3DShape()
842 "Depth of input (", input_depth_value, in Conv3DShape()
845 if (input_depth_value != filter_input_depth_value) { in Conv3DShape()
846 int64_t num_groups = input_depth_value / filter_input_depth_value; in Conv3DShape()