Searched refs:extent1 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | common.h | 214 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast() local
215 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()
218 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()
220 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
255 const int extent1 = extended_input1_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast() local
256 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()
259 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()
261 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 495 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast() local
496 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()
499 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()
501 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
536 const int extent1 = extended_input1_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast() local
537 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()
540 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()
542 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
|
| /external/jemalloc_new/include/jemalloc/internal/ |
| D | extent_inlines.h | 24 extent_lock2(tsdn_t *tsdn, extent_t *extent1, extent_t *extent2) { in extent_lock2() argument
25 assert(extent1 != NULL && extent2 != NULL); in extent_lock2()
26 mutex_pool_lock2(tsdn, &extent_mutex_pool, (uintptr_t)extent1, in extent_lock2()
31 extent_unlock2(tsdn_t *tsdn, extent_t *extent1, extent_t *extent2) { in extent_unlock2() argument
32 assert(extent1 != NULL && extent2 != NULL); in extent_unlock2()
33 mutex_pool_unlock2(tsdn, &extent_mutex_pool, (uintptr_t)extent1, in extent_unlock2()
|
| /external/jemalloc_new/src/ |
| D | extent.c | 134 extent_t *extent1 = rtree_leaf_elm_extent_read(tsdn, &extents_rtree, in extent_rtree_leaf_elm_try_lock() local
137 if (extent1 == NULL) { in extent_rtree_leaf_elm_try_lock()
144 extent_lock(tsdn, extent1); in extent_rtree_leaf_elm_try_lock()
148 if (extent1 == extent2) { in extent_rtree_leaf_elm_try_lock()
149 *result = extent1; in extent_rtree_leaf_elm_try_lock()
152 extent_unlock(tsdn, extent1); in extent_rtree_leaf_elm_try_lock()
|